CUDA error: device-side assert triggered

omarsou / layoutlm_CORD

Evaluation of the Layoutlm model on the CORD dataset

MIT License

32 stars 8 forks source link

CUDA error: device-side assert triggered #3

Open hamxahbhatti opened 3 years ago

hamxahbhatti commented 3 years ago

I am using your provided layoutlm notebook for training the dataset that you have shared but when i am doing training i am getting this error of CUDA error: device-side assert triggered at 0 epochs. Can you please upload the latest working code. Thanks.

GK4444 commented 2 years ago

did u use generate_dataset.ipynb, because in that file while creating pickle files the bbox were added in [x1, y3, x3, y1] format which was throwing the error, when changed to [x1,y1,x3,y3] works for me.