Update codec checkpoint config

Signed-off-by: Ante Jukić <[email protected]>
NVIDIA · Nov 9, 2023 · a2f521e · a2f521e
1 parent d49b73c
commit a2f521e
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 6 deletions.
diff --git a/examples/tts/conf/audio_codec/audio_codec_24000.yaml b/examples/tts/conf/audio_codec/audio_codec_24000.yaml
@@ -161,10 +161,17 @@ trainer:
 exp_manager:
   exp_dir: null
   name: ${name}
-  create_tensorboard_logger: true
-  create_checkpoint_callback: true
-  create_wandb_logger: false
+  create_tensorboard_logger: false
+  create_wandb_logger: true
+  wandb_logger_kwargs:
+    name: null
+    project: null
+  create_checkpoint_callback: true 
   checkpoint_callback_params:
     monitor: val_loss
+    mode: min
+    save_top_k: 5
+    save_best_model: true
+    always_save_nemo: true
   resume_if_exists: false
   resume_ignore_no_checkpoint: false
diff --git a/examples/tts/conf/audio_codec/encodec_24000.yaml b/examples/tts/conf/audio_codec/encodec_24000.yaml
@@ -161,10 +161,17 @@ trainer:
 exp_manager:
   exp_dir: null
   name: ${name}
-  create_tensorboard_logger: true
-  create_checkpoint_callback: true
-  create_wandb_logger: false
+  create_tensorboard_logger: false
+  create_wandb_logger: true
+  wandb_logger_kwargs:
+    name: null
+    project: null
+  create_checkpoint_callback: true 
   checkpoint_callback_params:
     monitor: val_loss
+    mode: min
+    save_top_k: 5
+    save_best_model: true
+    always_save_nemo: true
   resume_if_exists: false
   resume_ignore_no_checkpoint: false