Refactor baselines: add generic trainer and common utils (#153)
* refactor and add generic trainer class * fix to pass tests * change BaseModel to BaseTrainer * fix tensorboard import causing CI failure * modify circle-ci test script accordingly * doc, typing and other changes * rename BASELINE to TRAINER * merge from upstream master * Update Habitat-API to allow for no rendering sensors (#139) Update Habitat-API to allow for no rendering sensors * Added installation requirements step for sim installation in CI setup. (#159) Added installation requirements step for sim installation in CI setup * move RolloutStorage to utils * add environments.py * make ckpt dir if not exist * small fixes according to comments * changes according to comments * update readme * fix old config compatibility * fix missed isort lint
Showing
- .circleci/config.yml 1 addition, 1 deletion.circleci/config.yml
- configs/baselines/ppo.yaml 1 addition, 1 deletionconfigs/baselines/ppo.yaml
- habitat_baselines/README.md 10 additions, 48 deletionshabitat_baselines/README.md
- habitat_baselines/__init__.py 5 additions, 0 deletionshabitat_baselines/__init__.py
- habitat_baselines/agents/ppo_agents.py 1 addition, 1 deletionhabitat_baselines/agents/ppo_agents.py
- habitat_baselines/agents/slam_agents.py 9 additions, 9 deletionshabitat_baselines/agents/slam_agents.py
- habitat_baselines/common/base_trainer.py 52 additions, 0 deletionshabitat_baselines/common/base_trainer.py
- habitat_baselines/common/baseline_registry.py 64 additions, 0 deletionshabitat_baselines/common/baseline_registry.py
- habitat_baselines/common/env_utils.py 104 additions, 0 deletionshabitat_baselines/common/env_utils.py
- habitat_baselines/common/environments.py 83 additions, 0 deletionshabitat_baselines/common/environments.py
- habitat_baselines/common/rollout_storage.py 26 additions, 233 deletionshabitat_baselines/common/rollout_storage.py
- habitat_baselines/common/tensorboard_utils.py 32 additions, 20 deletionshabitat_baselines/common/tensorboard_utils.py
- habitat_baselines/common/utils.py 166 additions, 0 deletionshabitat_baselines/common/utils.py
- habitat_baselines/config/default.py 70 additions, 29 deletionshabitat_baselines/config/default.py
- habitat_baselines/config/pointnav/ppo.yaml 39 additions, 0 deletionshabitat_baselines/config/pointnav/ppo.yaml
- habitat_baselines/config/pointnav/ppo_train_test.yaml 12 additions, 0 deletionshabitat_baselines/config/pointnav/ppo_train_test.yaml
- habitat_baselines/evaluate_ppo.py 0 additions, 356 deletionshabitat_baselines/evaluate_ppo.py
- habitat_baselines/rl/ppo/__init__.py 1 addition, 2 deletionshabitat_baselines/rl/ppo/__init__.py
- habitat_baselines/rl/ppo/policy.py 1 addition, 1 deletionhabitat_baselines/rl/ppo/policy.py
- habitat_baselines/rl/ppo/ppo_trainer.py 575 additions, 0 deletionshabitat_baselines/rl/ppo/ppo_trainer.py
Loading
Please register or sign in to comment