# ------------------- General Options -------------------------
description                  : LED
results_root_dir             : results
dataset                      : nba

# ------------------- Dataset -------------------------
past_frames                  : 10
future_frames                : 20
min_past_frames              : 10
min_future_frames            : 20

motion_dim                   : 2
forecast_dim                 : 2

traj_mean                    : [14, 7.5]
traj_scale                   : 5

# ------------------- Model -------------------------
pretrained_core_denoising_model: './results/checkpoints/base_diffusion_model.p'
debug                        : False # set to True for early stop in each epoch.

diffusion                    : {
  steps                      : 100,
  beta_start                 : 1.e-4,
  beta_end                   : 5.e-2,
  beta_schedule              : 'linear'
}

# ------------------- Training Parameters -------------------------
lr                           : 1.e-3
train_batch_size             : 10
test_batch_size              : 500
num_epochs                   : 100

lr_scheduler                 : 'step'
decay_step                   : 8
decay_gamma                  : 0.5
test_interval                : 4