PyTorch implementation of Vanilla Policy Gradient, Truncated Natural Policy Gradient, Trust Region Policy Optimization, Proximal Policy Optimization.
- algorithm: PG, NPG, TRPO, PPO
- env: Ant-v2, HalfCheetah-v2, Hopper-v2, Humanoid-v2, HumanoidStandup-v2, InvertedPendulum-v2, Reacher-v2, Swimmer-v2, Walker2d-v2
- The system trains a Humanoid-v2 env using the PPO algorithm by default. If you wanna specify other envs and algs use:
python main.py --algorithm "algorithm name" --env "environment name"
This code is modified version of codes