Getting Nan after first iteration with custom loss

21

You can directly print these tensors in the forward pass to get their values for debugging.