It seems I was passing in the wrong thing to my loss function. I changed this line loss = criterion(outputs, torch.max(labels,1)[1]).
確定! 回上一頁