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 GitHub - ice-tong/pytorch-captcha: 使用pytorch搭建cnn识别验证码, thank u

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