lr 1e-3 -> 1e-4

Felix660 · Jan 27, 2021 · 8a4e17d · 8a4e17d
1 parent 6029154
commit 8a4e17d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -16,7 +16,7 @@
 parser.add_argument('--valid_length', type=float, default=2)
 parser.add_argument('--test_length', type=float, default=1)
 parser.add_argument('--epoch', type=int, default=50)
-parser.add_argument('--lr', type=float, default=1e-3)
+parser.add_argument('--lr', type=float, default=1e-4)
 parser.add_argument('--multi_layer', type=int, default=5)
 parser.add_argument('--device', type=str, default='cpu')
 parser.add_argument('--validate_freq', type=int, default=1)