TypeError: iteration over a 0-d tensor
.item()で取り出されたtensorの次元が0のために発生したエラーのため、下記の9から12行目のようにif c.ndim == 0で条件分岐させて対応。
with torch.no_grad(): for data in test_loader: inputs, labels = data outputs = net(inputs) _, predicted = torch.max(outputs, 1) c = (predicted == labels).squeeze() for i in range(len(labels)): label = labels if c.ndim == 0: class_correct += c.item() else: class_correct += c.item() class_total += 1