GPU training error!

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!

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

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