metadata
license: apache-2.0
pipeline_tag: reinforcement-learning
datasets:
- cheryyunl/Make-An-Agent
Pretrained models for Make-An-Agent. See https://github.com/cheryyunl/Make-An-Agent for usage instructions.
autoencoder.pt and behavior_embedding.pt are trained with parameter data and trajectory data in train_data/data.pt of our dataset.
model-best.pt is the policy generator model trained by latent parameter representation data and behavior embedding data in train_data/process_data.pt
Bibtex citation
@misc{liang2024makeanagentgeneralizablepolicynetwork,
title={Make-An-Agent: A Generalizable Policy Network Generator with Behavior-Prompted Diffusion},
author={Yongyuan Liang and Tingqiang Xu and Kaizhe Hu and Guangqi Jiang and Furong Huang and Huazhe Xu},
year={2024},
eprint={2407.10973},
archivePrefix={arXiv},
primaryClass={cs.AI},
url={https://arxiv.org/abs/2407.10973},
}