I would like to run the model on my own GPU, not TPU!
I would like to run the model on my own GPU, not TPU!