Lesson1.ipynb - Error - TypeError: torch.max received an invalid combination of arguments - got (numpy.ndarray, dim=int)

Part 1 (2017)
19

try using probs instead of log_preds in the accuracy_np function. Also, you should probably get the mean of probs, like in here

1 Like