Decrease default lr and add to docs on steps

theosanderson · Mar 2, 2023 · f8bbbbe · f8bbbbe
1 parent 7f45464
commit f8bbbbe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/chronumental/__main__.py b/src/chronumental/__main__.py
@@ -93,10 +93,10 @@ def get_parser():
     parser.add_argument('--steps',
                         default=1000,
                         type=int,
-                        help="Number of steps to use for the SVI")
+                        help="Number of steps to use for the SVI. Increasing this number will make runtime increase, but yield more accurate results.")
 
     parser.add_argument('--lr',
-                        default=0.1,
+                        default=0.05,
                         type=float,
                         help="Adam learning rate")