fix

2023-02-28 20:04:30 +00:00
parent d5e2c629b7
commit 4c6497f726
1 changed files with 3 additions and 0 deletions
@@ -880,6 +880,9 @@ if __name__ == '__main__':
            train_accumulator.reset()
            valid_accumulator.reset()

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

        step += 1