bug fix

2024-10-16 10:45:03 +08:00 · 2024-10-16 10:45:03 +08:00 · 768cdb1acc
parent 155585271a
commit 768cdb1acc
1 changed files with 1 additions and 0 deletions
--- a/CIFAR10_playground.py
+++ b/CIFAR10_playground.py
@ -260,6 +260,7 @@ for epoch in range(n_epochs):
    correct = 0
    total = 0
    for idx,(img,label) in tqdm(enumerate(test_loader)):
+        img, label=img.to(device), label.to(device)
        output = model(img)
        loss = criterion(output, label)
        valid_loss += loss.item() * img.shape[0]