Update chain inference

cfg/exp/chain_inference.yaml

@@ -26,8 +26,6 @@ effects_to_remove:
 datamodule:
   batch_size: 16
   num_workers: 8
-  train_dataset: None
-  val_dataset: None
 
 ckpts:
   RandomPedalboardDistortion: "ckpts/distortion.ckpt"

cfg/exp/chain_inference_aug.yaml

@@ -0,0 +1,42 @@
+# @package _global_
+defaults:
+  - override /model: demucs
+  - override /effects: all
+seed: 12345
+sample_rate: 48000
+chunk_size: 262144 # 5.5s
+logs_dir: "./logs"
+render_files: False
+render_root: "/scratch/EffectSet"
+accelerator: "gpu"
+log_audio: True
+# Effects
+num_kept_effects: [0,0] # [min, max]
+num_removed_effects: [0,5] # [min, max]
+shuffle_kept_effects: True
+shuffle_removed_effects: True
+num_classes: 5
+effects_to_keep:
+effects_to_remove:
+  - distortion
+  - compressor
+  - reverb
+  - chorus
+  - delay
+datamodule:
+  batch_size: 16
+  num_workers: 8
+
+ckpts:
+  RandomPedalboardDistortion: "ckpts/distortion_aug.ckpt"
+  RandomPedalboardCompressor: "ckpts/compressor_aug.ckpt"
+  RandomPedalboardReverb: "ckpts/reverb_aug.ckpt"
+  RandomPedalboardChorus: "ckpts/chorus_aug.ckpt"
+  RandomPedalboardDelay: "ckpts/delay_aug.ckpt"
+inference_effects_ordering:
+  - "RandomPedalboardDistortion"
+  - "RandomPedalboardCompressor"
+  - "RandomPedalboardReverb"
+  - "RandomPedalboardChorus"
+  - "RandomPedalboardDelay"
+num_bins: 1025