From d80556a462fbc65d5817ffbd77f22f210600999c Mon Sep 17 00:00:00 2001
From: "sweep-nightly[bot]"
 <131841235+sweep-nightly[bot]@users.noreply.github.com>
Date: Thu, 2 Nov 2023 22:29:33 +0000
Subject: [PATCH 1/2] feat: Updated src/main.py

---
 src/main.py | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/src/main.py b/src/main.py
index 243a31e..38f8db3 100644
--- a/src/main.py
+++ b/src/main.py
@@ -1,3 +1,4 @@
+import logging
 from PIL import Image
 import torch
 import torch.nn as nn
@@ -6,6 +7,8 @@
 from torch.utils.data import DataLoader
 import numpy as np
 
+logging.basicConfig(filename='training.log', level=logging.ERROR)
+
 # Step 1: Load MNIST Data and Preprocess
 transform = transforms.Compose([
     transforms.ToTensor(),
@@ -41,8 +44,15 @@ def forward(self, x):
     for images, labels in trainloader:
         optimizer.zero_grad()
         output = model(images)
-        loss = criterion(output, labels)
-        loss.backward()
-        optimizer.step()
+        try:
+            loss = criterion(output, labels)
+        except Exception as e:
+            logging.error("Error calculating loss", exc_info=True)
+            continue
+        try:
+            loss.backward()
+            optimizer.step()
+        except Exception as e:
+            logging.error("Error in backpropagation or optimization step", exc_info=True)
 
 torch.save(model.state_dict(), "mnist_model.pth")
\ No newline at end of file

From 09f98a47b32f10c155402892155ca79a8cc44dad Mon Sep 17 00:00:00 2001
From: "sweep-nightly[bot]"
 <131841235+sweep-nightly[bot]@users.noreply.github.com>
Date: Thu, 2 Nov 2023 22:30:19 +0000
Subject: [PATCH 2/2] feat: Updated requirements.txt

---
 requirements.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 4668e12..1942038 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -33,7 +33,7 @@ starlette==0.27.0
 sympy==1.12
 tomli==2.0.1
 tomlkit==0.12.1
-torch==2.1.0+cpu
+torch==1.10.0
 torchaudio==2.1.0+cpu
 torchvision==0.16.0+cpu
 typing_extensions==4.8.0