From 5e2e9392f8aaad4b3028038cf1df4900b9465c56 Mon Sep 17 00:00:00 2001
From: Antoine <caillon@ircam.fr>
Date: Tue, 25 May 2021 10:15:18 +0200
Subject: [PATCH] update gitignore

---
 .gitignore  |  6 +++++-
 config.yaml | 12 ++++++------
 train.py    |  2 +-
 3 files changed, 12 insertions(+), 8 deletions(-)

diff --git a/.gitignore b/.gitignore
index 2eec01f..9d096d2 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,2 +1,6 @@
 *build*
-*DS_Store
\ No newline at end of file
+*DS_Store
+*runs*
+*__pycache__*
+export/*
+preprocessed/*
\ No newline at end of file
diff --git a/config.yaml b/config.yaml
index 79083de..3c0b2ad 100644
--- a/config.yaml
+++ b/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
diff --git a/train.py b/train.py
index 5d0b792..7499035 100644
--- a/train.py
+++ b/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)