add frcrn model

toolbox/torchaudio/models/frcrn/unet.py

@@ -306,6 +306,7 @@ class UNet(nn.Module):
         :return:
         """
         x = inputs
+        print(f"inputs: {x.shape}")
 
         # go down
         xs = list()
@@ -315,19 +316,18 @@ class UNet(nn.Module):
             xs.append(x)
             # print(f"x: {x.shape}")
             x = encoder_layer.forward(x)
-            # print(f"x: {x.shape}")
+            print(f"x: {x.shape}")
             xs_se.append(x)
 
         # x shape: [b, c, 1, t', 2]
         x = self.fsmn.forward(x)
         # x shape: [b, c, 1, t', 2]
-        # print(f"fsmn")
+        print(f"fsmn")
 
         p = x
         for i, decoder_layers in enumerate(self.decoder_layers):
-            # print(f"x: {x.shape}")
             p = decoder_layers.forward(p)
-            # print(f"p: {p.shape}")
+            print(f"p: {p.shape}")
             if i == self.model_length - 1:
                 break
             p = torch.cat(tensors=[p, xs_se[self.model_length - 1 - i]], dim=1)