[hotfix] salesforce#449 (salesforce#450)

maulikmadhavi · Jul 24, 2023 · 61989d7 · 61989d7
1 parent 7fe1dd5
commit 61989d7
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/lavis/runners/runner_base.py b/lavis/runners/runner_base.py
@@ -360,11 +360,12 @@ def train(self):
             # training phase
             if not self.evaluate_only:
                 logging.info("Start training")
-                if cur_epoch == self.start_epoch:
-                    self.task.before_training(
-                        model=self.unwrap_dist_model(self.model),
-                        dataset=self.datasets["train"],
-                    )
+                # See https://github.com/salesforce/LAVIS/issues/449
+                # if cur_epoch == self.start_epoch:
+                #     self.task.before_training(
+                #         model=self.unwrap_dist_model(self.model),
+                #         dataset=self.datasets["train"],
+                #     )
                 train_stats = self.train_epoch(cur_epoch)
                 self.log_stats(split_name="train", stats=train_stats)