From fadca1e1821c640ae10309a75bfc9427d7eab347 Mon Sep 17 00:00:00 2001 From: sichu Date: Fri, 25 Oct 2024 15:55:08 +0000 Subject: [PATCH] remove __ost_init__ --- sub-packages/bionemo-llm/src/bionemo/llm/data/datamodule.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/sub-packages/bionemo-llm/src/bionemo/llm/data/datamodule.py b/sub-packages/bionemo-llm/src/bionemo/llm/data/datamodule.py index a99c73cfd..a4b0cbf9a 100644 --- a/sub-packages/bionemo-llm/src/bionemo/llm/data/datamodule.py +++ b/sub-packages/bionemo-llm/src/bionemo/llm/data/datamodule.py @@ -23,8 +23,9 @@ class MegatronDataModule(pl.LightningDataModule): """A mixin that adds a `state_dict` and `load_state_dict` method for datamodule training resumption in NeMo.""" - def __post_init__(self): + def _init__(self, *args, **kwargs): """Set init_global_step to 0 for datamodule resumption.""" + super().__init__(self, *args, **kwargs) self.init_global_step = 0 def update_init_global_step(self):