├── .gitignore ├── README.md ├── gail ├── __init__.py ├── config │ ├── CartPole-v0 │ │ ├── config_gail.json │ │ ├── config_ppo.json │ │ └── config_traj.json │ └── LunarLander-v2 │ │ ├── config_gail.json │ │ ├── config_ppo.json │ │ └── config_traj.json ├── main.py ├── model_args.py ├── ppo.py ├── traj.py ├── utils.py └── visualize.py ├── pretrained ├── CartPole-v0 │ ├── gail │ │ ├── discriminator.ckpt │ │ ├── policy.ckpt │ │ ├── record1.pkl │ │ ├── record2.pkl │ │ ├── record3.pkl │ │ ├── record4.pkl │ │ ├── record5.pkl │ │ └── rewards_gail.png │ ├── ppo │ │ ├── policy.ckpt │ │ ├── record1.pkl │ │ ├── record2.pkl │ │ ├── record3.pkl │ │ ├── record4.pkl │ │ ├── record5.pkl │ │ └── rewards_ppo.png │ └── trajectory │ │ ├── actions.csv │ │ └── states.csv └── LunarLander-v2 │ └── ppo │ └── policy.ckpt ├── report.pdf ├── requirements.txt └── setup.py /.gitignore: -------------------------------------------------------------------------------- 1 | **/wandb/ 2 | *.egg-info/ 3 | **/__pycache__/ 4 | .idea/ 5 | out/ -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /gail/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gail/config/CartPole-v0/config_gail.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/gail/config/CartPole-v0/config_gail.json -------------------------------------------------------------------------------- /gail/config/CartPole-v0/config_ppo.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/gail/config/CartPole-v0/config_ppo.json -------------------------------------------------------------------------------- /gail/config/CartPole-v0/config_traj.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/gail/config/CartPole-v0/config_traj.json -------------------------------------------------------------------------------- /gail/config/LunarLander-v2/config_gail.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/gail/config/LunarLander-v2/config_gail.json -------------------------------------------------------------------------------- /gail/config/LunarLander-v2/config_ppo.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/gail/config/LunarLander-v2/config_ppo.json -------------------------------------------------------------------------------- /gail/config/LunarLander-v2/config_traj.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/gail/config/LunarLander-v2/config_traj.json -------------------------------------------------------------------------------- /gail/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/gail/main.py -------------------------------------------------------------------------------- /gail/model_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/gail/model_args.py -------------------------------------------------------------------------------- /gail/ppo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/gail/ppo.py -------------------------------------------------------------------------------- /gail/traj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/gail/traj.py -------------------------------------------------------------------------------- /gail/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/gail/utils.py -------------------------------------------------------------------------------- /gail/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/gail/visualize.py -------------------------------------------------------------------------------- /pretrained/CartPole-v0/gail/discriminator.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/pretrained/CartPole-v0/gail/discriminator.ckpt -------------------------------------------------------------------------------- /pretrained/CartPole-v0/gail/policy.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/pretrained/CartPole-v0/gail/policy.ckpt -------------------------------------------------------------------------------- /pretrained/CartPole-v0/gail/record1.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/pretrained/CartPole-v0/gail/record1.pkl -------------------------------------------------------------------------------- /pretrained/CartPole-v0/gail/record2.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/pretrained/CartPole-v0/gail/record2.pkl -------------------------------------------------------------------------------- /pretrained/CartPole-v0/gail/record3.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/pretrained/CartPole-v0/gail/record3.pkl -------------------------------------------------------------------------------- /pretrained/CartPole-v0/gail/record4.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/pretrained/CartPole-v0/gail/record4.pkl -------------------------------------------------------------------------------- /pretrained/CartPole-v0/gail/record5.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/pretrained/CartPole-v0/gail/record5.pkl -------------------------------------------------------------------------------- /pretrained/CartPole-v0/gail/rewards_gail.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/pretrained/CartPole-v0/gail/rewards_gail.png -------------------------------------------------------------------------------- /pretrained/CartPole-v0/ppo/policy.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/pretrained/CartPole-v0/ppo/policy.ckpt -------------------------------------------------------------------------------- /pretrained/CartPole-v0/ppo/record1.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/pretrained/CartPole-v0/ppo/record1.pkl -------------------------------------------------------------------------------- /pretrained/CartPole-v0/ppo/record2.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/pretrained/CartPole-v0/ppo/record2.pkl -------------------------------------------------------------------------------- /pretrained/CartPole-v0/ppo/record3.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/pretrained/CartPole-v0/ppo/record3.pkl -------------------------------------------------------------------------------- /pretrained/CartPole-v0/ppo/record4.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/pretrained/CartPole-v0/ppo/record4.pkl -------------------------------------------------------------------------------- /pretrained/CartPole-v0/ppo/record5.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/pretrained/CartPole-v0/ppo/record5.pkl -------------------------------------------------------------------------------- /pretrained/CartPole-v0/ppo/rewards_ppo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/pretrained/CartPole-v0/ppo/rewards_ppo.png -------------------------------------------------------------------------------- /pretrained/CartPole-v0/trajectory/actions.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/pretrained/CartPole-v0/trajectory/actions.csv -------------------------------------------------------------------------------- /pretrained/CartPole-v0/trajectory/states.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/pretrained/CartPole-v0/trajectory/states.csv -------------------------------------------------------------------------------- /pretrained/LunarLander-v2/ppo/policy.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/pretrained/LunarLander-v2/ppo/policy.ckpt -------------------------------------------------------------------------------- /report.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/report.pdf -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jatinarora2702/gail-pytorch/HEAD/setup.py --------------------------------------------------------------------------------