working on trainer docs

2026-09-09 11:32:07 +08:00 · 2020-01-16 10:14:29 -05:00
parent 80454d3635
commit 1b12cb83dc
1 changed files with 4 additions and 1 deletions
@@ -355,7 +355,10 @@ class Trainer(TrainerIOMixin,
                    # if checkpoint callback used, then overrides the weights path
                    # **NOTE: this saves weights to some/path NOT my/path
                    checkpoint_callback = ModelCheckpoint(filepath='some/path')
-                    trainer = Trainer(checkpoint_callback=checkpoint_callback, weights_save_path='my/path')
+                    trainer = Trainer(
+                        checkpoint_callback=checkpoint_callback, 
+                        weights_save_path='my/path'
+                    )
                
        # :param str amp_level: Check nvidia docs for level
        """