How to verify if the tensorflow code trains completely in FP16?
How to verify if the tensorflow code trains completely in FP16? Question: I’m trying to train a TensorFlow (version 2.11.0) code in float16. I checked that FP16 is supported on the RTX 3090 GPU. So, I followed the below link to train the whole code in reduced precision. https://www.tensorflow.org/api_docs/python/tf/keras/backend/set_floatx Then, When I was testing the …
Total answers: 1