Thom Lane 81bac050d7 Added Custom Initialisation for MXNet Heads (#86) ...

* Added NormalizedRSSInitializer, using same method as TensorFlow backend, but changed name since ‘columns’ have different meaning in dense layer weight matrix in MXNet.

* Added unit test for NormalizedRSSInitializer.

2018-11-16 08:15:43 -08:00

..

__init__.py

Adding mxnet components to rl_coach/architectures (#60)

2018-11-07 17:07:15 +02:00

test_head.py

Added Custom Initialisation for MXNet Heads (#86)

2018-11-16 08:15:43 -08:00

test_ppo_head.py

Adding mxnet components to rl_coach/architectures (#60)

2018-11-07 17:07:15 +02:00

test_ppo_v_head.py

Adding mxnet components to rl_coach/architectures (#60)

2018-11-07 17:07:15 +02:00

test_q_head.py

Adding mxnet components to rl_coach/architectures (#60)

2018-11-07 17:07:15 +02:00

test_v_head.py

Adding mxnet components to rl_coach/architectures (#60)

2018-11-07 17:07:15 +02:00