AssertionError loss_value is NaN

[aman-autophoto-ai]

Hello,

I encountered an AssertionError while training a model, indicating that the loss_value is NaN. I'd like to get some assistance or insights into why this might be happening and how I can resolve it.

Environment:
OS: Ubuntu 22.04
Python Version: 3.6.13
Framework/Library Version: TensorFlow-GPU 1.15.0
Cuda: 10.0.130
Cudnn: 7.6.5
GPU: NVIDIA RTX 3070 TI
CPU: Intel i7