Add configuration files

saga.yml

@@ -0,0 +1,48 @@
+seed_everything: null
+trainer:
+  logger: true
+  default_root_dir: "results/"
+  max_epochs: 751
+  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: 35
+
+data:
+  dropout: 0.4
+  seqlen: 5
+  n_lookahead: 20
+  data_root: data/saga/processed/
+  x_channels: 63
+  cond_channels: 1017
+  framerate: 25
+  num_plot_samples: 5
+  is_training: true
+  num_workers: 16
+  trinity: false

trinity.yml

@@ -0,0 +1,48 @@
+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