PyTorch error (value mismatch) running prediction on cityscapes

Reagan1311 / DABNet

Depth-wise Asymmetric Bottleneck for Real-time Semantic Segmentation (BMVC2019)

https://github.com/Reagan1311/DABNet

MIT License

140 stars 36 forks source link

Closed niskov closed 5 years ago

niskov commented 5 years ago

Running the following command:

python predict.py --checkpoint DABNet_cityscapes.pth

I received the following error:

ValueError: non-broadcastable output operand with shape () doesn't match the broadcast shape (3,)