Dimensions not working
gulzainali98 opened this issue · 5 comments
Dimensions don't match up.
/usr/local/lib/python2.7/dist-packages/torch/nn/functional.py:1474: UserWarning: Using a target size (torch.Size([1966080])) that is different to the input size (torch.Size([3010560])) is deprecated. Please ensure they have the same size.
"Please ensure they have the same size.".format(target.size(), input.size()))
I am using BCELoss and i have tried with 192*192 dimensions as well and it is not working.
I am using this network for binary classification
Is the error from this script? https://github.com/usuyama/pytorch-unet/blob/master/pytorch_unet.py
@gulzainali98 friendly ping - great if you can help me locating the issue.
Hey sorry for late response,
This code works like a charm. It was my own mistake somewhere.
Great!