Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

rmv print

This commit is contained in:
shahules786 2022-09-29 16:05:02 +05:30
parent fd52625441
commit a0f70010f2
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
enhancer/models/waveunet.py
View File

@ -125,7 +125,6 @@ class WaveUnet(Model):
for layer,decoder in enumerate(self.decoders): for layer,decoder in enumerate(self.decoders):
out = F.interpolate(out, scale_factor=2, mode="linear") out = F.interpolate(out, scale_factor=2, mode="linear")
print(out.shape,encoder_outputs[layer].shape)
out = self.fix_last_dim(out,encoder_outputs[layer]) out = self.fix_last_dim(out,encoder_outputs[layer])
out = torch.cat([out,encoder_outputs[layer]],dim=1) out = torch.cat([out,encoder_outputs[layer]],dim=1)
out = decoder(out) out = decoder(out)