train

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