ZYM/PPCA-AIPacMan-2024
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ml q3 improve to 98.4%

Browse Source
This commit is contained in:
ZhuangYumin
2024-07-05 19:42:55 +08:00
parent 99f53727ba
commit 819fe60a6b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
machinelearning/models.py
View File

@ -298,8 +298,8 @@ class DigitClassificationModel(Module):
""" """
""" YOUR CODE HERE """ """ YOUR CODE HERE """
optimizer = torch.optim.Adam(self.parameters(), lr=0.0005) optimizer = torch.optim.Adam(self.parameters(), lr=0.0005)
dataloader = DataLoader(dataset, batch_size=20, shuffle=True) dataloader = DataLoader(dataset, batch_size=100, shuffle=True)
max_round=15000 max_round=30000
required_accuracy=0.99 required_accuracy=0.99
round_cnt=0 round_cnt=0
while round_cnt<max_round: while round_cnt<max_round: