search

BIT-MCS / Cadre

[AAAI 2022] CADRE: A Cascade Deep Reinforcement Learning Framework for Vision-based Autonomous Urban Driving
MIT License
26 stars 2 forks source link

hi, how can I retrain the CoPM? I can find the training manuscript and the dataset. #4

Closed hankerbit closed 11 months ago

nuomizai commented 11 months ago

Sorry for the trouble. We could not upload the training datasets of CoPM because they exceed Google Drive's storage capacity. However, we provide a sample dataset, and you can use it to try running the training code for CoPM with the script /scripts/train_encoder.sh.

We could also provide the very original code we used for collecting the datasets. You can find them under /statics (i.e., the vae_agent.py and the base_vae_agent.py.

hankerbit commented 11 months ago

Thanks very much, I find them. Nice work!!