As a 10708 PGM project, this repo contains implementations of
- VQVAE
- Discrete Diffusion
- Continuous Diffusion
- Linear/Quadratic Noise Schedules
- Heun's 2nd Order Method
- models
- diffusion:
continous_new.py
: contains the two architectures we mentioned in the reportcontinuous.py
diffusion_utils.py
diffusion_transformers.py
diffusion_continuous.py
: this is the main class to initialize the trajectory prediction networksvqvae.py
: contains VQVAE
- diffusion:
mkdir carla
cd carla
wget https://carla-releases.s3.eu-west-3.amazonaws.com/Linux/CARLA_0.9.10.1.tar.gz
wget https://carla-releases.s3.eu-west-3.amazonaws.com/Linux/AdditionalMaps_0.9.10.1.tar.gz
tar -xf CARLA_0.9.10.1.tar.gz
tar -xf AdditionalMaps_0.9.10.1.tar.gz
rm CARLA_0.9.10.1.tar.gz
rm AdditionalMaps_0.9.10.1.tar.gz
cd ..
# without display
SDL_VIDEODRIVER=offscreen SDL_HINT_CUDA_DEVICE=0 ./carla/CarlaUE4.sh --world-port=2000 -opengl
Training the model
python scripts/leaderboard/wp_IL/train_diffusion_continuous.py gpu=[0,1,2,3]
Running the agent
python leaderboard/scripts/run_evaluation.py user=$USER experiments=diffusion eval=longest6