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

adding the missing export_onnx_graph parameter to task parameters (#73)

Browse Source
This commit is contained in:
Itai Caspi authored and GitHub committed 2018-11-08 12:52:42 +02:00
1 parent 8f0415b4cc
commit 83e0b09a6a
3 files changed
+12 -6
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

rl_coach/architectures/tensorflow_components/heads/ppo_head.py
+1 -1
View File
@@ -19,7 +19,7 @@ import tensorflow as tf
from rl_coach.architectures.tensorflow_components.layers import Dense
from rl_coach.architectures.tensorflow_components.heads.head import Head, normalized_columns_initializer
from rl_coach.base_parameters import AgentParameters
from rl_coach.base_parameters import AgentParameters, DistributedTaskParameters
from rl_coach.core_types import ActionProbabilities
from rl_coach.spaces import BoxActionSpace, DiscreteActionSpace
from rl_coach.spaces import SpacesDefinition