c

Aquila-f · Aug 3, 2021 · b6d7b9d · b6d7b9d
1 parent ffb49b1
commit b6d7b9d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lab3/01.py b/lab3/01.py
@@ -121,10 +121,10 @@ def forward(self, x):
 
 config = {
     'Batch_size' : 4,
-    'Epochs' : 10,
+    'Epochs' : 50,
     'Optimizer' : 'SGD',
     'Optim_hparas':{
-        'lr' : 0.001,
+        'lr' : 0.0008,
         'momentum' : 0.9,
         'weight_decay' : 5e-4
     },
@@ -141,7 +141,7 @@ def forward(self, x):
 df_loss = pd.DataFrame()
 
 
-for switch in [True,False]:
+for switch in [True]:
 
     train_accuracy_list = []
     train_loss_list = []