When I train the model, training freezes at loss.backward() without any error message or warnings. It just freeze… I have print out the loss, and have attached one of the example here
loss tensor(0.6277, device='cuda:0', grad_fn=<BinaryCrossEntropyBackward>)
I am wondering does anyone have faced the same problem and is there any suggestions to solve it?
Thank you!!