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

View File

@ -125,7 +125,6 @@ class WaveUnet(Model):
for layer,decoder in enumerate(self.decoders):
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 = torch.cat([out,encoder_outputs[layer]],dim=1)
out = decoder(out)