Add more experiments

cfg/exp/chorus_only.yaml

@@ -0,0 +1,24 @@
+# @package _global_
+defaults:
+  - override /model: demucs
+  - override /effects: all
+seed: 12345
+sample_rate: 48000
+chunk_size: 262144 # 5.5s
+logs_dir: "./logs"
+render_files: True
+render_root: "/scratch/EffectSet"
+accelerator: "gpu"
+log_audio: True
+# Effects
+num_kept_effects: [0,0] # [min, max]
+num_removed_effects: [1,1] # [min, max]
+shuffle_kept_effects: True
+shuffle_removed_effects: False
+num_classes: 1
+effects_to_keep:
+effects_to_remove:
+  - chorus
+datamodule:
+  batch_size: 16
+  num_workers: 8

cfg/exp/compression_only.yaml

@@ -0,0 +1,24 @@
+# @package _global_
+defaults:
+  - override /model: demucs
+  - override /effects: all
+seed: 12345
+sample_rate: 48000
+chunk_size: 262144 # 5.5s
+logs_dir: "./logs"
+render_files: True
+render_root: "/scratch/EffectSet"
+accelerator: "gpu"
+log_audio: True
+# Effects
+num_kept_effects: [0,0] # [min, max]
+num_removed_effects: [1,1] # [min, max]
+shuffle_kept_effects: True
+shuffle_removed_effects: False
+num_classes: 1
+effects_to_keep:
+effects_to_remove:
+  - compressor
+datamodule:
+  batch_size: 16
+  num_workers: 8

cfg/exp/delay_only.yaml

@@ -0,0 +1,24 @@
+# @package _global_
+defaults:
+  - override /model: demucs
+  - override /effects: all
+seed: 12345
+sample_rate: 48000
+chunk_size: 262144 # 5.5s
+logs_dir: "./logs"
+render_files: True
+render_root: "/scratch/EffectSet"
+accelerator: "gpu"
+log_audio: True
+# Effects
+num_kept_effects: [0,0] # [min, max]
+num_removed_effects: [1,1] # [min, max]
+shuffle_kept_effects: True
+shuffle_removed_effects: False
+num_classes: 1
+effects_to_keep:
+effects_to_remove:
+  - delay
+datamodule:
+  batch_size: 16
+  num_workers: 8

cfg/exp/distortion_only.yaml

@@ -0,0 +1,24 @@
+# @package _global_
+defaults:
+  - override /model: demucs
+  - override /effects: all
+seed: 12345
+sample_rate: 48000
+chunk_size: 262144 # 5.5s
+logs_dir: "./logs"
+render_files: True
+render_root: "/scratch/EffectSet"
+accelerator: "gpu"
+log_audio: True
+# Effects
+num_kept_effects: [0,0] # [min, max]
+num_removed_effects: [1,1] # [min, max]
+shuffle_kept_effects: True
+shuffle_removed_effects: False
+num_classes: 1
+effects_to_keep:
+effects_to_remove:
+  - distortion
+datamodule:
+  batch_size: 16
+  num_workers: 8

cfg/exp/reverb_only.yaml

@@ -0,0 +1,24 @@
+# @package _global_
+defaults:
+  - override /model: demucs
+  - override /effects: all
+seed: 12345
+sample_rate: 48000
+chunk_size: 262144 # 5.5s
+logs_dir: "./logs"
+render_files: True
+render_root: "/scratch/EffectSet"
+accelerator: "gpu"
+log_audio: True
+# Effects
+num_kept_effects: [0,0] # [min, max]
+num_removed_effects: [1,1] # [min, max]
+shuffle_kept_effects: True
+shuffle_removed_effects: False
+num_classes: 1
+effects_to_keep:
+effects_to_remove:
+  - reverb
+datamodule:
+  batch_size: 16
+  num_workers: 8

cfg/model/tcn.yaml

@@ -13,7 +13,7 @@ model:
     noutputs: 1
     nblocks: 20
     channel_growth: 0
-    channel_width: 64
+    channel_width: 256
     kernel_size: 7
     stack_size: 10
     dilation_growth: 2

remfx/models.py

@@ -30,7 +30,6 @@ class RemFXChainInference(pl.LightningModule):
             {
                 "SISDR": SISDRLoss(),
                 "STFT": MultiResolutionSTFTLoss(),
-                "FAD": FADLoss(sample_rate=sample_rate),
             }
         )
         self.sample_rate = sample_rate
@@ -141,7 +140,6 @@ class RemFX(pl.LightningModule):
             {
                 "SISDR": SISDRLoss(),
                 "STFT": MultiResolutionSTFTLoss(),
-                "FAD": FADLoss(sample_rate=sample_rate),
             }
         )
         # Log first batch metrics input vs output only once
@@ -222,6 +220,7 @@ class RemFX(pl.LightningModule):
                     prog_bar=True,
                     sync_dist=True,
                 )
+
         return loss