From 819fe60a6b6c8f17f0f012c7edb18324da25e61d Mon Sep 17 00:00:00 2001
From: ZhuangYumin <zhuangyumin@sjtu.edu.cn>
Date: Fri, 5 Jul 2024 19:42:55 +0800
Subject: [PATCH] ml q3 improve to 98.4%

---
 machinelearning/models.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/machinelearning/models.py b/machinelearning/models.py
index af9965f..4824e81 100644
--- a/machinelearning/models.py
+++ b/machinelearning/models.py
@@ -298,8 +298,8 @@ class DigitClassificationModel(Module):
         """
         """ YOUR CODE HERE """
         optimizer = torch.optim.Adam(self.parameters(), lr=0.0005)
-        dataloader = DataLoader(dataset, batch_size=20, shuffle=True)
-        max_round=15000
+        dataloader = DataLoader(dataset, batch_size=100, shuffle=True)
+        max_round=30000
         required_accuracy=0.99
         round_cnt=0
         while round_cnt<max_round: