diff --git a/rl_coach/coach.py b/rl_coach/coach.py index 8c61c10..1994c68 100644 --- a/rl_coach/coach.py +++ b/rl_coach/coach.py @@ -425,7 +425,7 @@ class CoachLauncher(object): help="(flag) Render environment", action='store_true') parser.add_argument('-f', '--framework', - help="(string) Neural network framework. Available values: tensorflow", + help="(string) Neural network framework. Available values: tensorflow, mxnet", default='tensorflow', type=str) parser.add_argument('-n', '--num_workers',