update gitignore

.gitignore

@@ -1,2 +1,6 @@
 *build*
-*DS_Store
+*DS_Store
+*runs*
+*__pycache__*
+export/*
+preprocessed/*

config.yaml

@@ -1,22 +1,22 @@
 data:
-  data_location: "./data/"
+  data_location: /slow-2/georges/dataset-urmp-wav
   extension: "wav"
 
 preprocess:
-  sampling_rate: &samplingrate 48000
-  signal_length: &signallength 192000
-  block_size: &blocksize 512 # must be a power of 2 if using realtime
+  sampling_rate: &samplingrate 16000
+  signal_length: &signallength 64000
+  block_size: &blocksize 160 # must be a power of 2 if using realtime
 
   oneshot: false # crop every audio file to exactly signal length
   out_dir: ./preprocessed/
 
 model:
   hidden_size: 512
-  n_harmonic: 64
+  n_harmonic: 101
   n_bands: 65
   sampling_rate: *samplingrate
   block_size: *blocksize
 
 train:
-  scales: [4096,2048, 1024, 512, 256, 128]
+  scales: [4096, 2048, 1024, 512, 256, 128]
   overlap: .75

train.py

@@ -106,7 +106,7 @@ class args(Config):
         n_element += 1
         mean_loss += (loss.item() - mean_loss) / n_element
 
-    if not e % 1000:
+    if not e % 10:
         writer.add_scalar("lr", schedule(e), e)
         writer.add_scalar("reverb_decay", model.reverb.decay.item(), e)
         writer.add_scalar("reverb_wet", model.reverb.wet.item(), e)