Epoch issue WARNING:matplotlib.image:Clipping input data to the valid range for imshow with RGB data ([0..1] for floats or [0..255] for integers)

I am training my model loop where it was asked to assign epoch by self so the mse is around 320.

So first I tried 5 epoch, which gave me way higher mse, so I went next for 100 epoch, it was going fine till 98 epoch and then the below image error image happened as mse became nan. at 98 epoch mse was around 487. what could be going wrong here. please see the image below


r2FMI6e6mZmUIKX3LbA93Zg.png)

1 Like

I have a few suggestions for you. Maybe you should check your learning rate again, which might be too high. You can also take a look at the sampling layer and encoder layer where the VAE is located

1 Like

it is very old posted question, the issue has been resolved and it was epoch issue only

1 Like