gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 11:10:20 +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

6
tutorials/1. Implementing an Algorithm.ipynb
View File

@@ -363,9 +363,9 @@
"if not os.path.exists(log_path):\n",
" os.makedirs(log_path)\n",
" \n",
"task_parameters = TaskParameters(framework_type=\"tensorflow\", \n",
" evaluate_only=False,\n",
" experiment_path=log_path)\n",
"task_parameters = TaskParameters(framework_type=Frameworks.tensorflow, \n",
" evaluate_only=False,\n",
" experiment_path=log_path)\n",
"\n",
"task_parameters.__dict__['checkpoint_save_secs'] = None\n",
"\n",

10
tutorials/2. Adding an Environment.ipynb
View File

@@ -1,8 +1,10 @@
{
"cells": [
{
"cell_type": "markdown",
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"In this tutorial we'll add the DeepMind Control Suite environment to Coach, and create a preset that trains the DDPG agent on the new environment."
]
@@ -341,9 +343,9 @@
"if not os.path.exists(log_path):\n",
" os.makedirs(log_path)\n",
" \n",
"task_parameters = TaskParameters(framework_type=\"tensorflow\", \n",
" evaluate_only=False,\n",
" experiment_path=log_path)\n",
"task_parameters = TaskParameters(framework_type=Frameworks.tensorflow, \n",
" evaluate_only=False,\n",
" experiment_path=log_path)\n",
"\n",
"task_parameters.__dict__['checkpoint_save_secs'] = None\n",
"\n",

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",