HLPS

This is a PyTorch implementation for our paper "Probabilistic Subgoal Representations for Hierarchical Reinforcement learning" (ICML 2024).

Dependencies

• Python 3.7
• PyTorch 2.0
• OpenAI Gym 0.22
• MuJoCo 2.1.2

Run the codes with python train.py.

Citation

If you find this work useful in your research, please consider citing the paper:

@inproceedings{vhwang2024,
  title={Probabilistic Subgoal Representations for Hierarchical Reinforcement learning},
  author={Vivienne Huiling Wang and
          Tinghuai Wang and
          Wenyan Yang and
          Joni{-}Kristian K{\"{a}}m{\"{a}}r{\"{a}}inen and
          Joni Pajarinen},
  booktitle={ICML},
  year={2024}
}