gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +01:00
Code Issues Projects Releases Wiki Activity

Added MXNet to arg docs. (#121)

Browse Source
This commit is contained in:
Thom Lane
2018-11-19 01:31:28 -08:00
committed by Gal Novik
parent d4d06aaea6
commit 9210909050
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rl_coach/coach.py
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',