| |
| |
| |
| |
| import torch |
|
|
| def compute_batch_accuracy(pred, label): |
| correct = (pred == label).sum() |
| return correct,label.size(0) |
|
|
| def compute_set_accuracy(model, test_loader): |
| device = torch.device("cuda" if torch.cuda.is_available() else "cpu") |
| correct = 0 |
| total = 0 |
| with torch.no_grad(): |
| for data in test_loader: |
| inputs, labels = data |
| |
| inputs = inputs.to(device) |
| labels = labels.to(device) |
| outputs = model(inputs) |
| |
| correct_batch, total_batch = compute_batch_accuracy(torch.argmax(outputs, dim=1), labels) |
| correct += correct_batch |
| total += total_batch |
| |
| return correct/total |
