trinity.yml

seed_everything: null
trainer:
  logger: true
  default_root_dir: "results/"
  max_epochs: 131
  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: 2
  
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: 80
  max_grad_clip: 5
  max_grad_norm: 100
  beta_1: 0.9
  beta_2: 0.999
  eps: 1.0e-08
  warmup_steps: 3000
  min_lr: 0.00015
  init_lr: 0.0015
  checkpoints_gap: 100000000
  scalar_log_gaps: 1
  save_videos_every_n_epochs: 10

data:
  dropout: 0.4
  seqlen: 5
  n_lookahead: 20
  data_root: data/trinity/processed/
  x_channels: 45
  cond_channels: 927
  framerate: 20
  num_plot_samples: 5
  is_training: true
  num_workers: 20
  trinity: true