Remove change to training_epochs

Signed-off-by: Fabrice Normandin <[email protected]>
mila-iqia · Jun 29, 2023 · 65e3822 · 65e3822
1 parent 5c66038
commit 65e3822
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/docs/examples/good_practices/checkpointing/README.rst b/docs/examples/good_practices/checkpointing/README.rst
@@ -146,8 +146,7 @@ repository.
    +
 
     def main():
-   -    training_epochs = 10
-   +    training_epochs = 5
+        training_epochs = 10
         learning_rate = 5e-4
         weight_decay = 1e-4
         batch_size = 128

diff --git a/docs/examples/good_practices/checkpointing/main.py b/docs/examples/good_practices/checkpointing/main.py
@@ -49,7 +49,7 @@ class RunState(TypedDict):
 
 
 def main():
-    training_epochs = 5
+    training_epochs = 10
     learning_rate = 5e-4
     weight_decay = 1e-4
     batch_size = 128