feat: Updated src/main.py

sweepai · Nov 26, 2023 · daebccc · daebccc
1 parent 79c5051
commit daebccc
Showing 1 changed file with 0 additions and 6 deletions.
diff --git a/src/main.py b/src/main.py
@@ -14,12 +14,6 @@
 ])
 
 trainset = datasets.MNIST('.', download=True, train=True, transform=transform)
-trainloader = DataLoader(trainset, batch_size=64, shuffle=True)
-
-# Step 2: Define the PyTorch Model
-class Net(nn.Module):
-trainloader = DataLoader(trainset, batch_size=64, shuffle=True)
-
 # Step 2: Define the PyTorch Model
 class Net(nn.Module):
     def __init__(self):