Hi,
by using torch.autograd.set_detect_anomaly(True) Pytorch return this error:
Function 'PowBackward0' returned nan values in its 0th output.

more specifically when I use pow:
_x = torch.where(_x <= 0.008856, 9.033 * _x, 1.16 * (_x+epsilon).pow(1 / 3) - 0.16)

Someone has some ideas of how to avoid this?