Why program exit after the backward funtion ,please help

I download the code and try to run it, but it exit after the backward funtion,can anyone tell me why and how to solve it,the project is https://github.com/ice-tong/pytorch-captcha, thank u

i think it may be my cuda’s problem,i use the cpu to train it,and it works