fixed cli help for steps per epoch

henighan · May 3, 2019 · 414ea6e · 414ea6e
1 parent 740b162
commit 414ea6e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -32,7 +32,7 @@ Options:
                                   [default: 3]
   --epochs INTEGER                Number of epochs  [default: 50]
   -steps, --steps_per_epoch INTEGER
-                                  Number of epochs  [default: 4000]
+                                  Number of steps per epoch  [default: 4000]
   -env, --env_name TEXT           Environment name  [default: Swimmer-v2]
   -hid, --hidden_sizes TEXT       Hidden sizes for actor and critic MLPs
                                   [default: (64,64)]

diff --git a/deeprl/manage.py b/deeprl/manage.py
@@ -41,7 +41,7 @@ def process_cli_kwargs(all_kwargs):
 @click.option('--epochs', default=50, help='Number of epochs',
               show_default=True)
 @click.option('--steps_per_epoch', '-steps', default=4000,
-              help='Number of epochs', show_default=True)
+              help='Number of steps per epoch', show_default=True)
 @click.option('--env_name', '-env', default=['Swimmer-v2'],
               help='Environment name', show_default=True, multiple=True)
 @click.option('--hidden_sizes', '-hid', default=['(64,64)'],