GPU training error!

nlp
1

Hello everyone, I have this run-time error when I try to train the model on gpu. Can someone point out where I did wrong? Thanks!

image
image2333×707 74.1 KB

2

Your notebook doesn’t show any errors so did you already solve the issue?

3

I think the code is currently in cpu mode, can you switch to GPU mode? Thanks! @ptrblck