Add rllib agent and rollout script
make repository a package This will enable the rllib stuff to run correctly, remove some weird path hacking, and is generally a better development strategy We batch rollout evaluations, like in the standard rollout script. Currently needs a docker container with at least 2.5 Gb RAM, otherwise it crashes
parent
626aaad9
No related branches found
No related tags found
Showing
- .gitattributes 2 additions, 0 deletions.gitattributes
- .gitignore 1 addition, 0 deletions.gitignore
- Dockerfile 1 addition, 0 deletionsDockerfile
- agents/rllib_batched_agent.py 66 additions, 0 deletionsagents/rllib_batched_agent.py
- envs/wrappers.py 1 addition, 1 deletionenvs/wrappers.py
- nethack_baselines/rllib/README.md 155 additions, 1 deletionnethack_baselines/rllib/README.md
- nethack_baselines/rllib/__init__.py 6 additions, 0 deletionsnethack_baselines/rllib/__init__.py
- nethack_baselines/rllib/config.yaml 130 additions, 0 deletionsnethack_baselines/rllib/config.yaml
- nethack_baselines/rllib/envs.py 104 additions, 0 deletionsnethack_baselines/rllib/envs.py
- nethack_baselines/rllib/models.py 709 additions, 0 deletionsnethack_baselines/rllib/models.py
- nethack_baselines/rllib/train.py 146 additions, 0 deletionsnethack_baselines/rllib/train.py
- nethack_baselines/rllib/util/__init__.py 5 additions, 0 deletionsnethack_baselines/rllib/util/__init__.py
- nethack_baselines/rllib/util/id_pairs.py 141 additions, 0 deletionsnethack_baselines/rllib/util/id_pairs.py
- nethack_baselines/rllib/util/loading.py 58 additions, 0 deletionsnethack_baselines/rllib/util/loading.py
- requirements.txt 1 addition, 0 deletionsrequirements.txt
- rollout_rllib.py 107 additions, 0 deletionsrollout_rllib.py
- setup.py 26 additions, 0 deletionssetup.py
- submission_config.py 3 additions, 0 deletionssubmission_config.py
Loading
Please register or sign in to comment