cherubicXN / hawp

Holistically-Attracted Wireframe Parsing [TPAMI'23] & [CVPR' 20]
MIT License
291 stars 51 forks source link

Segmentation fault (core dumped) #5

Closed www322 closed 4 years ago

www322 commented 4 years ago

Hi, thank you for your work. When I train with your code, I met 'Segmentation fault (core dumped)' error, can you help check the code, thank you!

cherubicXN commented 4 years ago

Could you post the training log?

www322 commented 4 years ago

Could you post the training log?

just begin... 2020-06-12 11:21:37,967 hawp INFO: Saving config into: outputs/hawp/config.yml 2020-06-12 11:21:55,088 hawp.trainer INFO: No checkpoint found. Initializing model from scratch Segmentation fault (core dumped)

cherubicXN commented 4 years ago

Any more information?

www322 commented 4 years ago

i fork your project, just change 'png' to 'jpg', no more other changes.

cherubicXN commented 4 years ago

What are your hardware devices and the CUDA version?

cherubicXN commented 4 years ago

Could you run the data loader without model forward?

www322 commented 4 years ago

Could you run the data loader without model forward?

My cuda is 8.0 before, now i change to cuda10.2, now i can run normally. Thank you very much!