Update checkpointer.py

vturrisi · Jun 18, 2024 · 6985b84 · 6985b84
1 parent ccdba0b
commit 6985b84
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/solo/utils/checkpointer.py b/solo/utils/checkpointer.py
@@ -174,8 +174,7 @@ def on_train_epoch_end(self, trainer: pl.Trainer, _):
         if epoch % self.frequency == 0:
             self.save(trainer)
 
-    def on_train_end(self, trainer: pl.
-Trainer, _):
+    def on_train_end(self, trainer: pl.Trainer, _):
         """Saves model at the end of training.
 
         Args: