fix

2023-02-28 20:01:10 +00:00
parent 764df45095
commit d5e2c629b7
1 changed files with 3 additions and 0 deletions
@@ -584,6 +584,9 @@ if __name__ == '__main__':
            train_accumulator.reset()
            valid_accumulator.reset()

+        if step % 5000 == 0:
+            manager.save(checkpoint_number=step)
+            
        print(step)

        step += 1