pytorch error: RuntimeError: value cannot be converted to type float without overflow: (0.00655336,-0.00…

Solution:
initialize network layers such as conv2d.

# Default Method
for m in model.modules():
    if isinstance(m, (nn.Conv2d, nn.Linear)):
        nn.init.xavier_uniform_(m.weight)