Getting run time error while trying to train with custom data - ESPNetv2 - segmentation

sacmehta / EdgeNets

This repository contains the source code of our work on designing efficient CNNs for computer vision

MIT License

411 stars 82 forks source link

Closed kunalsinha9 closed 5 years ago

kunalsinha9 commented 5 years ago

RunTimeError: 1only batches of spatial targets supported (non-empty 3D tensors) but got targets of size of size: : [1, 256, 512, 3]

sacmehta commented 5 years ago

Your target images need to be grayscale and not RGB.