[Modify] epochs 100 -> 20

drawcodeboy · drawcodeboy · commit ca2a289950bf · 2025-12-12T23:56:11.000+09:00
diff --git a/configs/test/convnet.yaml b/configs/test/convnet.yaml
@@ -20,4 +20,4 @@ hyperparameters:
 
 save:
   weights_path: saved/weights
-  weights_filename: convnet_mnist.epochs_100.pth
+  weights_filename: convnet_mnist.epochs_020.pth
diff --git a/configs/test/convnet2.yaml b/configs/test/convnet2.yaml
@@ -20,4 +20,4 @@ hyperparameters:
 
 save:
   weights_path: saved/weights
-  weights_filename: convnet2_mnist.epochs_100.pth
+  weights_filename: convnet2_mnist.epochs_020.pth
diff --git a/configs/train/convnet.yaml b/configs/train/convnet.yaml
@@ -20,7 +20,7 @@ hyperparameters:
   optim: AdamW
   lr: 0.0001
   loss_fn: cross-entropy
-  epochs: 100
+  epochs: 20
 
 save:
   name: convnet_mnist
diff --git a/configs/train/convnet2.yaml b/configs/train/convnet2.yaml
@@ -20,7 +20,7 @@ hyperparameters:
   optim: AdamW
   lr: 0.0001
   loss_fn: cross-entropy
-  epochs: 100
+  epochs: 20
 
 save:
   name: convnet2_mnist
