"lpips nan, mse nan" using A100

[solomon-ma]

I have trained the model using Nvidia A100 GPU, but the terminal give us the above result.

I already test the code using Nvidia 2080Ti, all is fine.

Could you give me some suggestions how to use A100 to train it?