Sina Afrooze 77fb561668 Added code to fall back to CPU if GPU not available. (#150) ...

- Code will also prune GPU list if more than available GPUs is requested.

2018-11-25 08:32:26 +02:00

..

embedders

Add tensor input type for arbitrary dimensional observation (#125)

2018-11-19 16:41:12 +02:00

heads

Adding checkpointing framework (#74)

2018-11-19 19:45:49 +02:00

middlewares

fixing the dropout rate code (#72)

2018-11-08 16:53:47 +02:00

__init__.py

Adding mxnet components to rl_coach/architectures (#60)

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

architecture.py

Added code to fall back to CPU if GPU not available. (#150)

2018-11-25 08:32:26 +02:00

general_network.py

Moved tf.variable_scope and tf.device calls to framework-specific architecture (#136)

2018-11-22 22:52:21 +02:00

layers.py

Adding mxnet components to rl_coach/architectures (#60)

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

savers.py

Adding checkpointing framework (#74)

2018-11-19 19:45:49 +02:00

utils.py

Moved tf.variable_scope and tf.device calls to framework-specific architecture (#136)

2018-11-22 22:52:21 +02:00