Closed madhavmk closed 4 years ago
Hi, thank you for issue. Try to squeeze(1).
Yes. using 'torch.squeeze(var, 1)' to return tensors with single dimensional quantities removed seems to be working. I will create a pull request with this fix once i verify that there are no errors.
Thanks for the clear code implementation of the Complex Unet paper !
Unfortunately I'm facing a Tensor Dimension RunTime error when training the DCUnet on both my local setup and on Colaboratory. I have pasted the error stack below.
A Runtime Error occurs when running Cell 28 : train_losses, test_losses = train(dcunet10, train_loader, test_loader, loss_fn, optimizer, scheduler, 3)
Any help on this would be appreciated. Thanks :)