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 ability to switch between tensorflow and mxnet using -f commandline argument. (#48)

Browse Source 
NOTE: tensorflow framework works fine if mxnet is not installed in env, but mxnet will not work if tensorflow is not installed because of the code in network_wrapper.
This commit is contained in:
Sina Afrooze
2018-10-30 15:29:34 -07:00
committed by Scott Leishman
parent 2046358ab0
commit 95b4fc6888
8 changed files with 47 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tutorials/3. Implementing a Hierarchical RL Graph.ipynb
View File

@@ -368,7 +368,7 @@
"if not os.path.exists(log_path):\n",
" os.makedirs(log_path)\n",
" \n",
"task_parameters = TaskParameters(framework_type=\"tensorflow\", \n",
"task_parameters = TaskParameters(framework_type=Frameworks.tensorflow, \n",
" evaluate_only=False,\n",
" experiment_path=log_path)\n",
"\n",