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

4
rl_coach/tests/agents/test_agent_external_communication.py
View File

@@ -1,7 +1,7 @@
import os
import sys
from rl_coach.base_parameters import TaskParameters
from rl_coach.base_parameters import TaskParameters, Frameworks
sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(__file__))))
import tensorflow as tf
@@ -16,7 +16,7 @@ def test_get_QActionStateValue_predictions():
from rl_coach.presets.CartPole_DQN import graph_manager as cartpole_dqn_graph_manager
assert cartpole_dqn_graph_manager
cartpole_dqn_graph_manager.create_graph(task_parameters=
TaskParameters(framework_type="tensorflow",
TaskParameters(framework_type=Frameworks.tensorflow,
experiment_path="./experiments/test"))
cartpole_dqn_graph_manager.improve_steps.num_steps = 1
cartpole_dqn_graph_manager.steps_between_evaluation_periods.num_steps = 5