Refactor PPO baseline to allow extensions (#144)
* Refactor baselines to allow for extension * Make clip an after-backward * Loop over rollout step instead of function for full rollout
Showing
- habitat_baselines/agents/ppo_agents.py 13 additions, 4 deletionshabitat_baselines/agents/ppo_agents.py
- habitat_baselines/common/rollout_storage.py 18 additions, 4 deletionshabitat_baselines/common/rollout_storage.py
- habitat_baselines/rl/models/__init__.py 0 additions, 0 deletionshabitat_baselines/rl/models/__init__.py
- habitat_baselines/rl/models/rnn_state_encoder.py 143 additions, 0 deletionshabitat_baselines/rl/models/rnn_state_encoder.py
- habitat_baselines/rl/models/simple_cnn.py 147 additions, 0 deletionshabitat_baselines/rl/models/simple_cnn.py
- habitat_baselines/rl/ppo/__init__.py 2 additions, 2 deletionshabitat_baselines/rl/ppo/__init__.py
- habitat_baselines/rl/ppo/policy.py 101 additions, 224 deletionshabitat_baselines/rl/ppo/policy.py
- habitat_baselines/rl/ppo/ppo.py 35 additions, 8 deletionshabitat_baselines/rl/ppo/ppo.py
- habitat_baselines/rl/ppo/ppo_trainer.py 115 additions, 79 deletionshabitat_baselines/rl/ppo/ppo_trainer.py
Loading
Please register or sign in to comment
