Beckschen/TransUNet

Hi!. When I am training the model, the cross entropy is not calculated. After that line of code, it produces cuda runtime error: device triggered. Does anyone know what is the reason for that ?

Opened this issue · 0 comments

Hi!. When I am training the model, the cross entropy is not calculated. After that line of code, it produces cuda runtime error: device triggered. Does anyone know what is the reason for that ?