diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..adbb97d
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+data/
\ No newline at end of file
diff --git a/dataset/__pycache__/cifar10.cpython-38.pyc b/dataset/__pycache__/cifar10.cpython-38.pyc
new file mode 100644
index 0000000..f1b4bf8
Binary files /dev/null and b/dataset/__pycache__/cifar10.cpython-38.pyc differ
diff --git a/dataset/__pycache__/cifar10.cpython-39.pyc b/dataset/__pycache__/cifar10.cpython-39.pyc
new file mode 100644
index 0000000..6dd1ae8
Binary files /dev/null and b/dataset/__pycache__/cifar10.cpython-39.pyc differ
diff --git a/freeze b/freeze
new file mode 100644
index 0000000..e69de29
diff --git a/models/__pycache__/wideresnet.cpython-38.pyc b/models/__pycache__/wideresnet.cpython-38.pyc
new file mode 100644
index 0000000..8d4362e
Binary files /dev/null and b/models/__pycache__/wideresnet.cpython-38.pyc differ
diff --git a/models/__pycache__/wideresnet.cpython-39.pyc b/models/__pycache__/wideresnet.cpython-39.pyc
new file mode 100644
index 0000000..e5782e0
Binary files /dev/null and b/models/__pycache__/wideresnet.cpython-39.pyc differ
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..a9c4fc5
--- /dev/null
+++ 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
index 5c7905e..f7a825e 100644
--- 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
new file mode 100644
index 0000000..bbad009
Binary files /dev/null and b/utils/__pycache__/__init__.cpython-38.pyc differ
diff --git a/utils/__pycache__/__init__.cpython-39.pyc b/utils/__pycache__/__init__.cpython-39.pyc
new file mode 100644
index 0000000..4439313
Binary files /dev/null and b/utils/__pycache__/__init__.cpython-39.pyc differ
diff --git a/utils/__pycache__/eval.cpython-38.pyc b/utils/__pycache__/eval.cpython-38.pyc
new file mode 100644
index 0000000..b6a07b6
Binary files /dev/null and b/utils/__pycache__/eval.cpython-38.pyc differ
diff --git a/utils/__pycache__/eval.cpython-39.pyc b/utils/__pycache__/eval.cpython-39.pyc
new file mode 100644
index 0000000..e8d5cf2
Binary files /dev/null and b/utils/__pycache__/eval.cpython-39.pyc differ
diff --git a/utils/__pycache__/logger.cpython-38.pyc b/utils/__pycache__/logger.cpython-38.pyc
new file mode 100644
index 0000000..f3b6dc3
Binary files /dev/null and b/utils/__pycache__/logger.cpython-38.pyc differ
diff --git a/utils/__pycache__/logger.cpython-39.pyc b/utils/__pycache__/logger.cpython-39.pyc
new file mode 100644
index 0000000..5a0bbde
Binary files /dev/null and b/utils/__pycache__/logger.cpython-39.pyc differ
diff --git a/utils/__pycache__/misc.cpython-38.pyc b/utils/__pycache__/misc.cpython-38.pyc
new file mode 100644
index 0000000..d5cd6dc
Binary files /dev/null and b/utils/__pycache__/misc.cpython-38.pyc differ
diff --git a/utils/__pycache__/misc.cpython-39.pyc b/utils/__pycache__/misc.cpython-39.pyc
new file mode 100644
index 0000000..3c8c274
Binary files /dev/null and b/utils/__pycache__/misc.cpython-39.pyc differ
diff --git a/utils/eval.py b/utils/eval.py
index 5051350..5b84c88 100644
--- 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
\ No newline at end of file
+    return res