updates to DER CNN notebook

deepskies · Jul 25, 2024 · 61b32ce · 61b32ce
1 parent 5d9d1af
commit 61b32ce
Show file tree

Hide file tree

Showing 2 changed files with 216 additions and 13 deletions.
diff --git a/notebooks/load_CNN_chk.ipynb b/notebooks/load_CNN_chk.ipynb
diff --git a/src/utils/defaults.py b/src/utils/defaults.py
@@ -25,9 +25,9 @@
         "init_lr": 0.001,
         "BETA": 0.5,
         "n_epochs": 100,
-        "save_all_checkpoints": False,
-        "save_final_checkpoint": False,
-        "overwrite_final_checkpoint": False,
+        "save_all_checkpoints": True,
+        "save_final_checkpoint": True,
+        "overwrite_final_checkpoint": True,
         "plot": False,
         "savefig": False,
         "save_chk_random_seed_init": False,
@@ -75,9 +75,9 @@
         "init_lr": 0.001,
         "COEFF": 0.01,
         "n_epochs": 100,
-        "save_all_checkpoints": False,
-        "save_final_checkpoint": False,
-        "overwrite_final_checkpoint": False,
+        "save_all_checkpoints": True,
+        "save_final_checkpoint": True,
+        "overwrite_final_checkpoint": True,
         "plot": False,
         "savefig": False,
         "save_chk_random_seed_init": False,