| description : LED — SDD (Stanford Drone Dataset) | |
| results_root_dir : results_sdd | |
| dataset : sdd | |
| past_frames : 8 | |
| future_frames : 12 | |
| motion_dim : 2 | |
| forecast_dim : 2 | |
| traj_mean : [0.0, 0.0] | |
| traj_scale : 1.0 | |
| per_scene_norm : false | |
| pretrained_core_denoising_model : './results_sdd/checkpoints/base_diffusion_model_sdd.p' | |
| diffusion: | |
| steps : 100 | |
| beta_start : 1.e-4 | |
| beta_end : 5.e-2 | |
| beta_schedule : 'linear' | |
| train_batch_size : 32 | |
| test_batch_size : 256 | |
| num_epochs : 100 | |
| decay_step : 8 | |
| decay_gamma : 0.5 | |
| test_interval : 4 | |
| pretrain: | |
| num_epochs : 150 | |
| lr : 1.e-3 | |
| train_batch_size : 64 | |
| test_batch_size : 256 | |
| decay_step : 30 | |
| decay_gamma : 0.5 | |