update

toolbox/torchaudio/models/spectrum_dfnet/modeling_spectrum_dfnet.py

@@ -838,11 +838,11 @@ class SpectrumDfNet(nn.Module):
             raise AssertionError
 
         # spec_m = self.mask.forward(spec, mask)
-        #
-        # # lsnr shape: [batch_size, time_steps, 1]
-        # lsnr = torch.transpose(lsnr, dim0=2, dim1=1)
-        # # lsnr shape: [batch_size, 1, time_steps]
-        #
+
+        # lsnr shape: [batch_size, time_steps, 1]
+        lsnr = torch.transpose(lsnr, dim0=2, dim1=1)
+        # lsnr shape: [batch_size, 1, time_steps]
+
         # df_coefs = self.df_decoder.forward(emb, c0)
         # df_coefs = self.df_out_transform(df_coefs)
         # # df_coefs shape: [batch_size, df_order, time_steps, df_bins, 2]