Error when training encoders

[rafael-alencar29]

I get this error whenever I try to train the encoders (BottleneckAutoencoder, BaselineAutoencoder...)
"RuntimeError: Function 'PowBackward0' returned nan values in its 0th output"
I believe all the setting are right, I'm running on Colab.

Am I missing something?

[viettr]

Can you please post the whole error message? And does the error occur in the first epoch or later?

[rafael-alencar29]

I'm sharing the whole file link.

The error happens in the first epoch.
The the others models ran just fine, just the encoders I'm having problems.

https://colab.research.google.com/drive/1bWydVwUPWTncH9gGEt5Bm3s2K_XE8Wmo?usp=sharing

[rafael-alencar29]

Here is the error screenshot in case the link above doesn't work.