Fix TCN

cfg/model/dptnet.yaml

@@ -10,13 +10,10 @@ model:
   network:
     _target_: remfx.models.DPTNetModel
     n_src: 1
-    in_chan: 1
-    out_chan: 1
     bn_chan: 128
     hid_size: 128
     chunk_size: 100
     n_repeats: 6
-    win_len: 2
     fb_name: "free"
     kernel_size: 16
     n_filters: 1

cfg/model/tcn.yaml

@@ -11,10 +11,10 @@ model:
     _target_: remfx.models.TCNModel
     ninputs: 1
     noutputs: 1
-    nblocks: 4
+    nblocks: 20
     channel_growth: 0
     channel_width: 32
-    kernel_size: 13
+    kernel_size: 7
     stack_size: 10
     dilation_growth: 10
     condition: False

remfx/tcn.py

@@ -128,10 +128,7 @@ class TCN(nn.Module):
         x_in = x
         for _, block in enumerate(self.process_blocks):
             x = block(x)
-        # y_hat = torch.tanh(self.output(x))
-        x_in = causal_crop(x_in, x.shape[-1])
-        gain_ln = self.output(x)
-        y_hat = torch.tanh(gain_ln * x_in)
+        y_hat = torch.tanh(self.output(x))
         return y_hat
 
     def compute_receptive_field(self):

shell_vars.sh

@@ -1,3 +1,3 @@
-export DATASET_ROOT="./data/"
+export DATASET_ROOT="./data/remfx-data"
 export WANDB_PROJECT="RemFX"
 export WANDB_ENTITY="mattricesound"