search

zanilzanzan / FuseNet_PyTorch

Joint scene classification and semantic segmentation with FuseNet
GNU General Public License v3.0
108 stars 34 forks source link

Missing unsqueeze #13

Closed eliabntt closed 5 years ago

eliabntt commented 5 years ago

Without unsqueeze training will fail due to wrong dimensions of tensors