coach

gryf/coach

Fork 0

mirror of https://github.com/gryf/coach.git synced 2026-01-30 12:15:49 +01:00

Commit Graph

Author	SHA1	Message	Date
Thom Lane	7ba1a4393f	Channel order transpose, for image embedder. Updated unit test. (#87 )	2018-11-19 15:39:03 +02:00
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
Sina Afrooze	5fadb9c18e	Adding mxnet components to rl_coach/architectures (#60 ) Adding mxnet components to rl_coach architectures. - Supports PPO and DQN - Tested with CartPole_PPO and CarPole_DQN - Normalizing filters don't work right now (see #49) and are disabled in CartPole_PPO preset - Checkpointing is disabled for MXNet	2018-11-07 17:07:15 +02:00

Author

SHA1

Message

Date

Thom Lane

7ba1a4393f

Channel order transpose, for image embedder. Updated unit test. (#87 )

2018-11-19 15:39:03 +02:00

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

Sina Afrooze

5fadb9c18e

Adding mxnet components to rl_coach/architectures (#60 )

Adding mxnet components to rl_coach architectures.

- Supports PPO and DQN
- Tested with CartPole_PPO and CarPole_DQN
- Normalizing filters don't work right now (see #49) and are disabled in CartPole_PPO preset
- Checkpointing is disabled for MXNet

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

3 Commits