adding masking back
@@ -138,7 +138,7 @@ class Mask(nn.Module):
super(Mask, self).__init__()
def forward(self, mask, magnitude_spectrogram):
- return mask * magnitude_spectrogram
+ return mask * magnitude_spectrogram.unsqueeze(-1)
class RecurrentStack(nn.Module):
def __init__(self, num_features, hidden_size, num_layers, bidirectional, dropout, rnn_type='lstm'):
