You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
criterion=nn.CrossEntropyLoss()
print(outputs.data)
print(label.data)
loss=criterion(outputs, label) # getting error at this point
The output that I'm getting is =>
0.0174
0.1866
[torch.FloatTensor of size 2]
0
1
[torch.FloatTensor of size 2]
This is correct as far as the documentation is concern, but still getting following error dimension out of range (expected to be in range of [-1, 0], but got 1)
rathrio, hzshang, davidirobinson, shayanfazeli, aaronlelevier and 98 moreh-grieve, 4ndr3aR, eiskry, Raakeshm1738, Rosenn2000 and 1 more