chaslie:

how do i do that?


Add the print statements before the loss calculation via:
print(input.mean(), input.min(). input.max())
print(target.mean(), ...)

We could use these stats to create random tensors and try to run into this issue locally.
Also, could you check your numpy version via print(np.__version__), please?