fix args of model checkpoint callback

PTG-Kitware · Feb 25, 2021 · bcf4146 · bcf4146
1 parent 62781eb
commit bcf4146
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/project/configs/callbacks/default_callbacks.yaml b/project/configs/callbacks/default_callbacks.yaml
@@ -4,7 +4,8 @@ model_checkpoint:
     save_top_k: 2           # save k best models (determined by above metric)
     save_last: True         # additionaly always save model from last epoch
     mode: "max"             # can be "max" or "min"
-    filepath: "checkpoints/{epoch}"
+    dirpath: 'checkpoints/'
+    filename: 'sample-mnist-{epoch:02d}'
 
 
 early_stopping:

diff --git a/project/configs/datamodule/mnist_datamodule.yaml b/project/configs/datamodule/mnist_datamodule.yaml
@@ -2,5 +2,5 @@ _target_: src.datamodules.mnist_datamodule.MNISTDataModule
 data_dir: ${data_dir}  # data_dir is specified in config.yaml
 batch_size: 64
 train_val_test_split: [55_000, 5_000, 10_000]
-num_workers: 1
+num_workers: 0
 pin_memory: False
diff --git a/project/src/utils/template_utils.py b/project/src/utils/template_utils.py
@@ -93,4 +93,4 @@ def extras(config, model, datamodule, callbacks, loggers, trainer):
 
 def finish():
     wandb.finish()
-    neptune.stop()
+    # neptune.stop()