Cleaned up repo

IamShubhamGupto · Apr 14, 2023 · c3edb57 · c3edb57
1 parent 9d0a1c0
commit c3edb57
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 8 deletions.
diff --git a/train_100_old.py → archive/train_100_old.py b/train_100_old.py → archive/train_100_old.py
diff --git a/train_old.py → archive/train_old.py b/train_old.py → archive/train_old.py
diff --git a/train.py b/train.py
@@ -170,14 +170,8 @@
     train_loss, train_acc, n = 0, 0, 0
 
     #Go through training steps
-    for i, (X, y) in enumerate(trainloader):
-
-        print(type(X),print(type(y)))
-        print(X)
-        print(y)
-        print(X.shape,y.shape)
-
-
+    for i, (X, y) in enumerate(trainloader):  
+
         #Set train mode and port sample to cuda
         model.train()
         X, y = X.cuda(), y.cuda()