1
0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +01:00

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
This commit is contained in:
Sina Afrooze
2018-11-07 07:07:15 -08:00
committed by Itai Caspi
parent e7a91b4dc3
commit 5fadb9c18e
39 changed files with 3864 additions and 44 deletions

View File

@@ -68,6 +68,17 @@ if not using_GPU:
else:
install_requires.append('tensorflow-gpu==1.9.0')
# Framework-specific dependencies.
extras = {
'mxnet': ['mxnet-cu90mkl>=1.3.0']
}
all_deps = []
for group_name in extras:
all_deps += extras[group_name]
extras['all'] = all_deps
setup(
name='rl-coach',
version='0.10.0',
@@ -78,6 +89,7 @@ setup(
packages=find_packages(),
python_requires=">=3.5.*",
install_requires=install_requires,
extras_require=extras,
package_data={'rl_coach': ['dashboard_components/*.css',
'environments/doom/*.cfg',
'environments/doom/*.wad',