Skip to content

Commit dc7b964

Browse files
update default lr (#3002)
1 parent 4391c5a commit dc7b964

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/contrib/forecast/bart.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -171,7 +171,7 @@ def transform(pred, truth):
171171
parser.add_argument("--test-window", default=336, type=int)
172172
parser.add_argument("--stride", default=168, type=int)
173173
parser.add_argument("-n", "--num-steps", default=501, type=int)
174-
parser.add_argument("-lr", "--learning-rate", default=0.05, type=float)
174+
parser.add_argument("-lr", "--learning-rate", default=0.01, type=float)
175175
parser.add_argument("--dct", action="store_true")
176176
parser.add_argument("--num-samples", default=100, type=int)
177177
parser.add_argument("--log-every", default=50, type=int)

0 commit comments

Comments
 (0)