zdchu / MiLEt

Implementation of MiLEt for review

Code for AAAI 2024 long paper Meta-Reinforcement Learning via Exploratory Task Clustering. The code is partially adapted from the VariBAD implementation.

Usage

Run the model with default settings, for example, on the Ant-Goal environment

python main.py --env-type ant_goal_milet

Citation

Please cite our work if you find it useful to your research

@article{chu2023meta,
 title={Meta-Reinforcement Learning via Exploratory Task Clustering},
 author={Chu, Zhendong and Wang, Hongning},
 journal={arXiv preprint arXiv:2302.07958},
 year={2023}
}