search

RolandGao / RegSeg

The official implementation of "Rethink Dilated Convolution for Real-time Semantic Segmentation"
MIT License
86 stars 14 forks source link

can't train #2

Closed fathersdog closed 2 years ago

fathersdog commented 2 years ago

when I train the model on Cityscapes,I While get "RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED".

RolandGao commented 2 years ago

What's your output for after running the command nvidia-smi

fathersdog commented 2 years ago

Sorry,i've found the problem,it's my dataset generation error.