| data: |
| name: trajectory_nba_filling |
| full_seq_len: 30 |
| delta_len: 30 |
| num_agents: 11 |
| coord_dim: 2 |
| court_width: 94.0 |
| court_height: 50.0 |
| params: |
| train_batch_size: 128 |
| val_batch_size: 512 |
| full_seq_len: 30 |
| num_agents: 11 |
| coord_dim: 2 |
| court_width: 94.0 |
| court_height: 50.0 |
| train_path: data/nba_train.npy |
| val_path: data/nba_test.npy |
| trajectory_key: trajectory |
| context_key: context |
| mask_key: obs_mask |
| position_0_key: position_0 |
| context_fill: |
| - -4.0 |
| - -4.0 |
| include_delta_in_context: true |
| delta_context_fill: |
| - 0.0 |
| - 0.0 |
| delta_shift: |
| - 0.0 |
| - 0.0 |
| delta_scale: |
| - 1.0 |
| - 1.0 |
| masking: |
| train: |
| mixture: |
| - weight: 0.35 |
| even: |
| mode: random |
| prefix_min: 1 |
| prefix_max: 20 |
| - weight: 0.35 |
| agent: |
| mode: random |
| n_masked_min: 5 |
| n_masked_max: 11 |
| - weight: 0.3 |
| hybrid: |
| combine: union |
| a: |
| even: |
| mode: random |
| prefix_min: 1 |
| prefix_max: 20 |
| b: |
| agent: |
| mode: random |
| n_masked_min: 5 |
| n_masked_max: 11 |
| val_mask_path: /root/code/gameplay-trajectory-diffusion/data/nba_test_mask_v1.npy |
| num_workers: 0 |
| max_val_samples: 5120 |
| model: |
| name: trajectory_filling_ddpm |
| trajectory_key: trajectory |
| context_key: context |
| mask_key: obs_mask |
| position_0_key: position_0 |
| guidance_scale: 2.0 |
| p_uncond: 0.1 |
| log_blend_trajectory_video: false |
| timesteps: 1000 |
| beta_schedule: linear |
| linear_start: 0.0001 |
| linear_end: 0.02 |
| cosine_s: 0.008 |
| parameterization: eps |
| loss_type: l2 |
| diffusion_loss_type: rescaled_mse |
| log_diagnostic_vb: true |
| vb_decoder_nll: continuous |
| model_var_type: learned |
| clip_denoised: false |
| legacy_posterior_log_variance: false |
| backbone: |
| _target_: src.modules.backbones.dit_backbone.DITBackbone |
| max_seq_len: 30 |
| num_agents: 11 |
| coord_dim: 2 |
| context_channels: 2 |
| context_dim: 256 |
| n_temporal_layer: 4 |
| d_model_temporal: 256 |
| nhead_temporal: 8 |
| d_ff_temporal: 512 |
| n_spatial_layer: 4 |
| d_model_spatial: 256 |
| nhead_spatial: 8 |
| d_ff_spatial: 512 |
| num_timesteps: 1000 |
| patch_size: 5 |
| learn_sigma: true |
| seed: 42 |
| hardware: |
| use_gpu: true |
| gpu_devices: 1 |
| wandb: |
| enabled: true |
| project: trajectory-filling-ddpm |
| entity: null |
| name: dit-learnsigma-mixedmask-lr=1e-4 |
| save_dir: ./wandb |
| log_model: false |
| logging: |
| backend: null |
| tensorboard: |
| save_dir: ./tensorboard_logs |
| name: null |
| trainer: |
| lightning: |
| max_epochs: 1500 |
| accelerator: auto |
| devices: 1 |
| precision: 32 |
| log_every_n_steps: 50 |
| check_val_every_n_epoch: 20 |
| val_check_interval: null |
| limit_val_batches: null |
| deterministic: false |
| fast_dev_run: false |
| val_logging: |
| enabled: true |
| num_samples: 6 |
| log_every_n_val_epochs: 1 |
| optim: |
| learning_rate: 0.0001 |
| betas: |
| - 0.9 |
| - 0.999 |
| weight_decay: 0.0 |
| ema: |
| enabled: true |
| decay: 0.9999 |
| use_num_updates: true |
| checkpoint: |
| monitor: val/jade_min_30frames |
| mode: min |
| save_top_k: 4 |
| val_trajectory_metrics: |
| enabled: true |
| max_samples: 512 |
| every_n_val_epochs: 1 |
| num_paths: 20 |
| horizon_stride: 30 |
| metrics_start_t: 0 |
| prediction_mode: pure |
| guidance_scale_override: 2.0 |
| verbose: false |
| sampling: |
| method: dpm |
| dpm: |
| steps: 40 |
| order: 3 |
| skip_type: time_uniform |
| algorithm_type: dpmsolver++ |
| solver_method: singlestep |
| lower_order_final: true |
| denoise_to_zero: false |
| sampling: |
| method: ancestral |
|
|
