gryf/coach
1
0 Fork 0
mirror of https://github.com/gryf/coach.git synced 2026-10-13 18:22:29 +02:00
Code Issues Projects Releases Wiki Activity

Added MXNet to arg docs. (#121)

Browse Source
This commit is contained in:
Thom Lane authored and Gal Novik committed 2018-11-19 11:31:28 +02:00
1 parent d4d06aaea6
commit 9210909050
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
rl_coach/coach.py
+1 -1
View File
@@ -425,7 +425,7 @@ class CoachLauncher(object):
help="(flag) Render environment", help="(flag) Render environment",
action='store_true') action='store_true')
parser.add_argument('-f', '--framework', parser.add_argument('-f', '--framework',
help="(string) Neural network framework. Available values: tensorflow", help="(string) Neural network framework. Available values: tensorflow, mxnet",
default='tensorflow', default='tensorflow',
type=str) type=str)
parser.add_argument('-n', '--num_workers', parser.add_argument('-n', '--num_workers',