update default lr (#3002)

pyro-ppl · Dec 28, 2021 · dc7b964 · dc7b964
1 parent 4391c5a
commit dc7b964
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)