From 6a83ad7805da8e6733c8a0604070a7a7bc9fb4da Mon Sep 17 00:00:00 2001 From: martin jankowiak Date: Tue, 28 Dec 2021 16:27:54 -0500 Subject: [PATCH] update default lr --- examples/contrib/forecast/bart.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/contrib/forecast/bart.py b/examples/contrib/forecast/bart.py index fb7b1d9ee3..06e80c8d37 100644 --- a/examples/contrib/forecast/bart.py +++ b/examples/contrib/forecast/bart.py @@ -171,7 +171,7 @@ def transform(pred, truth): parser.add_argument("--test-window", default=336, type=int) parser.add_argument("--stride", default=168, type=int) parser.add_argument("-n", "--num-steps", default=501, type=int) - parser.add_argument("-lr", "--learning-rate", default=0.05, type=float) + parser.add_argument("-lr", "--learning-rate", default=0.01, type=float) parser.add_argument("--dct", action="store_true") parser.add_argument("--num-samples", default=100, type=int) parser.add_argument("--log-every", default=50, type=int)