not sure if tensorflow is running on GPU or CPU custom keras training loop or if its a problem with the training loop code

Question

the python script is using 20% of my CPU and only 0.3% of my GPU. I am still very new to Keras and TensorFlow so it is not at all unlikely that I am doing something wrong which