Update utils2.py

Kedreamix · Kedreamix · commit 0c3b20ada263 · 2022-09-07T16:23:27.000+08:00
diff --git a/CIFAR10_code/utils2.py b/CIFAR10_code/utils2.py
@@ -94,7 +94,7 @@ def train(epoch, epochs, model, dataloader, criterion, optimizer, scheduler = No
             loss.backward()
             optimizer.step()
                 
-            pbar.set_postfix(**{'Train Acc' : running_accuracy.item()/(step+1),
+            pbar.set_postfix(**{'Train Acc' : running_accuracy.items()/(step+1),
                                 'Train Loss' :running_loss.item()/(step+1)})
             pbar.update(1)
     if scheduler:
