search

VITA-Group / FasterSeg

[ICLR 2020] "FasterSeg: Searching for Faster Real-time Semantic Segmentation" by Wuyang Chen, Xinyu Gong, Xianming Liu, Qian Zhang, Yuan Li, Zhangyang Wang
MIT License
524 stars 107 forks source link

RuntimeError: transform: failed to synchronize: cudaErrorAssert: device-side assert triggered #63

Open PatrickHaopc opened 3 years ago

PatrickHaopc commented 3 years ago

Thanks for your work, I'm trying to use FasterSeg with my own custom data. My tack is just Two-class task. The imgsize of my data is 1280*640, and the integers in mask are 0 and 15, 0 represents background and 15 is the target. The "dataset files" are also done: image The num_classes I changed to 2, height and width are also adjusted. Error: image