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.
trafficstars

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