File "/media/data/Kaggle/Humpback_Whale_Identification/Models.py", line 57, in forward
    x = torch.sum(x ** self.p, dim=2) / (H * W)

RuntimeError: Function 'PowBackward1' returned nan values in its 1th output.
Useful thing! It seems the problem is in the first pow. Any ideas on how to fix it?