add frcrn model

examples/frcrn/yaml/config.yaml

@@ -16,14 +16,14 @@ seed: 1234
 
 sample_rate: 8000
 segment_size: 32000
-nfft: 512
-win_size: 512
-hop_size: 256
+nfft: 128
+win_size: 128
+hop_size: 64
 win_type: hann
 
 use_complex_networks: true
-model_depth: 20
-model_complexity: 45
+model_depth: 10
+model_complexity: -1
 
 min_snr_db: -10
 max_snr_db: 20

toolbox/torchaudio/models/frcrn/unet.py

@@ -340,13 +340,22 @@ class UNet(nn.Module):
 
 def main():
     # [batch_size, 1, freq_bins, time_steps, 2]
-    x = torch.rand(size=(1, 1, 257, 2000, 2))
-    # x = torch.rand(size=(1, 1, 256, 2000, 2))
-    # x = torch.rand(size=(1, 1, 255, 2000, 2))
+    # x = torch.rand(size=(1, 1, 257, 2000, 2))
+    # unet = UNet(
+    #     in_channels=1,
+    #     model_complexity=45,
+    #     model_depth=20,
+    #     use_complex_networks=True
+    # )
+    # print(unet)
+    # result = unet.forward(x)
+    # print(result.shape)
+
+    x = torch.rand(size=(1, 1, 65, 2000, 2))
     unet = UNet(
         in_channels=1,
-        model_complexity=45,
-        model_depth=20,
+        model_complexity=-1,
+        model_depth=10,
         use_complex_networks=True
     )
     print(unet)