fix

2023-02-28 20:00:09 +00:00
parent f479643be6
commit 6320275ea5
4 changed files with 17 additions and 13 deletions
@@ -568,6 +568,9 @@ if __name__ == '__main__':
            train_accumulator.reset()
            valid_accumulator.reset()

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

        step += 1