Kushaj: Thank you ! What did you find & change.
I found one issue with the way I had defined in my code
all_y_trues = torch.tensor([
1.,
0.,
0.,
1.])
I changed it to
all_y_trues = torch.tensor([
[1.],
[0.],
[0.],
[1.]])
it doesn’t explode now in the midway of epochs iteration