pytorch-classification
pytorch-classification copied to clipboard
RuntimeError: view size is not compatible with input tensor's size and stride (at least one dimension spans across two contiguous subspaces). Use .reshape(...) instead.

Hi, about this issue, you can with
correct_k = correct[:k].reshape(-1).float().sum(0)
to replaced
correct_k = correct[:k].view(-1).float().sum(0)
in utils/eval.py