3d model not using GPU

vision
1

I am currently using this 3d model in public kernel. This model is not using gpu even after moving model and input to available gpu. What could be reason…

https://www.kaggle.com/raviet/osic-final-easy-nn-using-image-and-tabular-data

@ptrblck any help i can get.

2

Are you sure you are using the GPU? What do you get when you print out the device variable? How do you know it’s not using the GPU?