From 9210909050c7d3a3ae9c3a9d38d18dee50e5ea64 Mon Sep 17 00:00:00 2001 From: Thom Lane Date: Mon, 19 Nov 2018 01:31:28 -0800 Subject: [PATCH] Added MXNet to arg docs. (#121) --- rl_coach/coach.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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',