Itai Caspi 811152126c Export graph to ONNX (#61) ...

Implements the ONNX graph exporting feature. 
Currently does not work for NAF, C51 and A3C_LSTM due to unsupported TF layers in the tf2onnx library.

2018-11-06 10:55:21 +02:00

..

__init__.py

Move embedder, middleware, and head parameters to framework agnostic modules. (#45)

2018-10-29 14:46:40 -07:00

categorical_q_head.py

Move embedder, middleware, and head parameters to framework agnostic modules. (#45)

2018-10-29 14:46:40 -07:00

cil_head.py

Move embedder, middleware, and head parameters to framework agnostic modules. (#45)

2018-10-29 14:46:40 -07:00

ddpg_actor_head.py

Move embedder, middleware, and head parameters to framework agnostic modules. (#45)

2018-10-29 14:46:40 -07:00

dnd_q_head.py

Move embedder, middleware, and head parameters to framework agnostic modules. (#45)

2018-10-29 14:46:40 -07:00

dueling_q_head.py

Move embedder, middleware, and head parameters to framework agnostic modules. (#45)

2018-10-29 14:46:40 -07:00

head.py

Move embedder, middleware, and head parameters to framework agnostic modules. (#45)

2018-10-29 14:46:40 -07:00

measurements_prediction_head.py

Move embedder, middleware, and head parameters to framework agnostic modules. (#45)

2018-10-29 14:46:40 -07:00

naf_head.py

Move embedder, middleware, and head parameters to framework agnostic modules. (#45)

2018-10-29 14:46:40 -07:00

policy_head.py

Move embedder, middleware, and head parameters to framework agnostic modules. (#45)

2018-10-29 14:46:40 -07:00

ppo_head.py

Export graph to ONNX (#61)

2018-11-06 10:55:21 +02:00

ppo_v_head.py

Move embedder, middleware, and head parameters to framework agnostic modules. (#45)

2018-10-29 14:46:40 -07:00

q_head.py

Move embedder, middleware, and head parameters to framework agnostic modules. (#45)

2018-10-29 14:46:40 -07:00

quantile_regression_q_head.py

Move embedder, middleware, and head parameters to framework agnostic modules. (#45)

2018-10-29 14:46:40 -07:00

rainbow_q_head.py

Move embedder, middleware, and head parameters to framework agnostic modules. (#45)

2018-10-29 14:46:40 -07:00

v_head.py

Move embedder, middleware, and head parameters to framework agnostic modules. (#45)

2018-10-29 14:46:40 -07:00