LPIENet， an error code : Floating point exception (core dumped)

[alexliyang]

LPIENet pytorch code, when change x = torch.ones(1, 3, 16, 16) to x = torch.ones(1, 3, 2048, 2048) , then run it , get error Floating point exception (core dumped), how to correct it?