Open vermavinay982 opened 1 year ago
Device side assert triggered, CUDA Runtime error NLL Loss forward reduce cuda kernel 2d.
While training with Prediction as "Attn" With CTC it is training perfectly.
Here is the config params for reference 11 class predictions- all numbers Sequence- BiLSTM Feature extraction- ResNet Transformation- TPS
Device side assert triggered, CUDA Runtime error NLL Loss forward reduce cuda kernel 2d.
While training with Prediction as "Attn" With CTC it is training perfectly.
Here is the config params for reference 11 class predictions- all numbers Sequence- BiLSTM Feature extraction- ResNet Transformation- TPS