YU1ut · ishnoor-singh2 · Sep 12, 2023 · Sep 12, 2023 · Sep 12, 2023 · Sep 12, 2023
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1 @@
+data/
diff --git a/dataset/__pycache__/cifar10.cpython-38.pyc b/dataset/__pycache__/cifar10.cpython-38.pyc
diff --git a/dataset/__pycache__/cifar10.cpython-39.pyc b/dataset/__pycache__/cifar10.cpython-39.pyc
diff --git a/freeze b/freeze
diff --git a/models/__pycache__/wideresnet.cpython-38.pyc b/models/__pycache__/wideresnet.cpython-38.pyc
diff --git a/models/__pycache__/wideresnet.cpython-39.pyc b/models/__pycache__/wideresnet.cpython-39.pyc
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,6 @@
+torch==1.4
+torchvision==0.2.2
+tensorboardX
+progress
+matplotlib
+numpy
diff --git a/train.py b/train.py
@@ -200,16 +200,18 @@ def train(labeled_trainloader, unlabeled_trainloader, model, optimizer, ema_opti
     model.train()
     for batch_idx in range(args.train_iteration):
         try:
-            inputs_x, targets_x = labeled_train_iter.next()
+            inputs_x, targets_x = next(labeled_train_iter)
         except:
             labeled_train_iter = iter(labeled_trainloader)
-            inputs_x, targets_x = labeled_train_iter.next()
+            inputs_x, targets_x = next(labeled_train_iter)
+
 
         try:
-            (inputs_u, inputs_u2), _ = unlabeled_train_iter.next()
+            (inputs_u, inputs_u2), _ = next(unlabeled_train_iter)
         except:
             unlabeled_train_iter = iter(unlabeled_trainloader)
-            (inputs_u, inputs_u2), _ = unlabeled_train_iter.next()
+            (inputs_u, inputs_u2), _ = next(unlabeled_train_iter)
+
 
         # measure data loading time
         data_time.update(time.time() - end)

diff --git a/utils/__pycache__/__init__.cpython-38.pyc b/utils/__pycache__/__init__.cpython-38.pyc
diff --git a/utils/__pycache__/__init__.cpython-39.pyc b/utils/__pycache__/__init__.cpython-39.pyc
diff --git a/utils/__pycache__/eval.cpython-38.pyc b/utils/__pycache__/eval.cpython-38.pyc
diff --git a/utils/__pycache__/eval.cpython-39.pyc b/utils/__pycache__/eval.cpython-39.pyc
diff --git a/utils/__pycache__/logger.cpython-38.pyc b/utils/__pycache__/logger.cpython-38.pyc
diff --git a/utils/__pycache__/logger.cpython-39.pyc b/utils/__pycache__/logger.cpython-39.pyc
diff --git a/utils/__pycache__/misc.cpython-38.pyc b/utils/__pycache__/misc.cpython-38.pyc
diff --git a/utils/__pycache__/misc.cpython-39.pyc b/utils/__pycache__/misc.cpython-39.pyc
diff --git a/utils/eval.py b/utils/eval.py
@@ -13,6 +13,6 @@ def accuracy(output, target, topk=(1,)):
 
     res = []
     for k in topk:
-        correct_k = correct[:k].view(-1).float().sum(0)
+        correct_k = correct[:k].reshape(-1).float().sum(0)
         res.append(correct_k.mul_(100.0 / batch_size))
-    return res
+    return res