ml q3 improve to 98.4%

2024-07-05 19:42:55 +08:00
parent 99f53727ba
commit 819fe60a6b
1 changed files with 2 additions and 2 deletions
--- 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: