Metrics

for data, target in test_loader:
    output   = model(data)
    pred     = output.argmax(dim=1, keepdim=True)
    correct += pred.eq(target.view_as(pred)).sum().item()

acc = 100. * correct / len(test_loader.dataset)
print('Accuracy: {acc}%')

---