train

2023-12-25 16:23:38 +08:00
parent d168f2946e
commit c756af358d
3 changed files with 9 additions and 3 deletions
@@ -3,4 +3,6 @@ test.py
 *.log
 Dataset/*
 *.ipynb
-*.swp
+*.swp
+.vscode/*
+
@@ -15,9 +15,13 @@
                "--lr",
                "10e-3",
                "--epochs",
-                "50",
+                "1000",
                "--batch_size",
-                "512",
+                "64",
+                "--if_scheduler",
+                "True",
+                "step_size",
+                "500e3",
                "--dataset",
                "cifar10",
                "--num_workers",