I am trying to train an autoencoder and I using torch.nn.MSELoss() as the loss function. I keep getting this error below:
RuntimeError                              Traceback (most recent call last)
<ipython-input-76-67910671b729> in <module>()
     20 
     21         optimizer.zero_grad()
---> 22         loss.backward()
     23         optimizer.step()
     24 
RuntimeError: Function 'MseLossBackward' returned nan values in its 0th output.

I have tried reading many of the other posts on this topic and I have tried various approaches such as adjusting the learning rate, gradient clipping. The weird thing is the loss doesn’t seem to diverge at all. I am not totally sure why this error keeps occurring for me, any suggestions or help would be appreciated? I am very new to pytorch.