saga.yml

seed_everything: null
trainer:
  logger: true
  default_root_dir: "results/"
  max_epochs: 1501
  max_time: null
  limit_train_batches: 1.0
  flush_logs_every_n_steps: 100
  log_every_n_steps: 50
  precision: 32
  weights_summary: top
  num_sanity_val_steps: 0
  
model:
  hidden_channels: 512
  K: 16
  actnorm_scale: 1.0
  flow_permutation: invconv
  flow_coupling: affine
  network_model: LSTM
  num_layers: 2
  LU_decomposed: true
  distribution: normal
  batch_size: 64
  max_grad_clip: 5
  max_grad_norm: 100
  lr_scheduler: none
  noam_warmup_steps: null
  noam_min_lr: null
  learning_rate: 0.0015
  beta_1: 0.9
  beta_2: 0.999
  eps: 1.0e-08
  save_videos_every_n_epochs: 50
  track_velocities_every_n_epochs: 30
  sampling_temp: 1.0
  
data:
  add_caption_to_vis: true
  dropout: 0.4
  seqlen: 5
  n_lookahead: 20
  data_root: data/saga/processed/
  x_channels: 39
  cond_channels: 923
  framerate: 25
  is_training: true
  num_workers: 4
  trinity: false