modified musdb training config so that it provides decent results

Eugene-Y · May 22, 2020 · 7630f29 · 7630f29
1 parent bd9cd3e
commit 7630f29
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/configs/musdb_config.json b/configs/musdb_config.json
@@ -10,17 +10,17 @@
     "T":512,
     "F":1024,
     "n_channels":2,
-    "n_chunks_per_song":1,
+    "n_chunks_per_song":40,
     "separation_exponent":2,
     "mask_extension":"zeros",
     "learning_rate": 1e-4,
     "batch_size":4,
     "training_cache":"cache/training",
     "validation_cache":"cache/validation",
-    "train_max_steps": 100000,
-    "throttle_secs":600,
+    "train_max_steps": 200000,
+    "throttle_secs":1800,
     "random_seed":3,
-    "save_checkpoints_steps":300,
+    "save_checkpoints_steps":1000,
     "save_summary_steps":5,
     "model":{
         "type":"unet.unet",