From c80fd1e101c1c4f20e85543edc7137567b4500a4 Mon Sep 17 00:00:00 2001
From: "sweep-nightly[bot]"
 <131841235+sweep-nightly[bot]@users.noreply.github.com>
Date: Tue, 24 Oct 2023 19:46:56 +0000
Subject: [PATCH] feat: Updated src/main.py

---
 src/main.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/main.py b/src/main.py
index b75a38b..a38d2ec 100644
--- a/src/main.py
+++ b/src/main.py
@@ -1,3 +1,9 @@
+import torch
+import torch.nn as nn
+import torch.optim as optim
+from torchvision import datasets, transforms
+from torch.utils.data import DataLoader
+
 class MNISTTrainer:
     def __init__(self):
         self.transform = transforms.Compose([