From 35586d6bca791a0c83fc33c2528b32bd3a7cc943 Mon Sep 17 00:00:00 2001
From: yayoimizuha <poko.peko.tomo@icloud.com>
Date: Sun, 16 Apr 2023 15:13:46 +0900
Subject: [PATCH] update

---
 resnet_finetune.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/resnet_finetune.py b/resnet_finetune.py
index f4974e2..845e0cd 100644
--- a/resnet_finetune.py
+++ b/resnet_finetune.py
@@ -95,7 +95,7 @@ for i in range(epochs):
     val_acc = .0
 
     model_gpu.train()
-    for images, labels in tqdm(dataloader['train'], leave=False):
+    for images, labels in dataloader['train']:
         optimizer.zero_grad()
         images = images.to(device)
         labels = labels.to(device)