feat: add seed setting for deterministic training

mnist_classifier/train.py

@@ -7,8 +7,22 @@ from mnist_classifier.dataset import MNISTDataModule
 from mnist_classifier.model import MNISTModel
 from datetime import datetime
 import os
+import random
+import numpy as np
+
+def set_seed(seed):
+    torch.manual_seed(seed)
+    torch.cuda.manual_seed(seed)
+    torch.cuda.manual_seed_all(seed)
+    np.random.seed(seed)
+    random.seed(seed)
+    torch.backends.cudnn.deterministic = True
+    torch.backends.cudnn.benchmark = False
 
 def train():
+    # Set seed for reproducibility
+    set_seed(42)
+    
     # Set device
     device = torch.device('cuda')
     print(f"Using device: {device}")

models/mnist_model_lr0.001_bs64_ep10.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1f00fa1ee4fd08e6a5c41d3952b64e27b8bb122182f432332e18c9ee2af67609
+oid sha256:d1b474acf8a447dea4e3aaaf0371346ee7a7055d1c716fb371c059b9a1799bab
 size 4803144