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)