avBuffer / UNet3plus_pth

UNet3+/ UNet++/UNet, used in Deep Automatic Portrait Matting in Pytorth
243 stars 39 forks source link

'tuple' object has no attribute 'size' #3

Open caixh39 opened 4 years ago

caixh39 commented 4 years ago

hi, I using "UNet3Plus_DeepSup" network to train, meeting a problem:"masks_pred = net(imgs)" in train.py,the masks_pred is a tuple which contains 5 values. then, I confusing how to compute "masks_pred " with "masks_truth" loss? thank you very much!

heifengtai commented 3 years ago

I also got the same problem. Please help me. Thank you very much.