configs/models/gsn_replica_config.yaml [4:26]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
  num_workers: 4
  batch_size: 16
  img_res: 64
  seq_len: 4
  step: 10
  depth: True
  center: 'mid'
  normalize_rotation: False
  rot_aug: False
  single_sample_per_trajectory: True
  train_set_config:
    split: 'train'
  val_set_config:
    split: 'train'  # eval on same distribution as training
    samples_per_epoch: 5000

model_config:
  target: models.gsn.GSN
  params:
    img_res: ???  # tied to img_res defined by data_config
    lr: 0.002
    ttur_ratio: 1.0  # controls ratio between G and D lr, where g_lr = lr and d_lr = lr * ttur_ratio
    voxel_res: 32
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



configs/models/gsn_vizdoom_config.yaml [4:26]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
  num_workers: 4
  batch_size: 16
  img_res: 64
  seq_len: 4
  step: 10
  depth: True
  center: 'mid'
  normalize_rotation: False
  rot_aug: False
  single_sample_per_trajectory: True
  train_set_config:
    split: 'train'
  val_set_config:
    split: 'train'  # eval on same distribution as training
    samples_per_epoch: 5000

model_config:
  target: models.gsn.GSN
  params:
    img_res: ???  # tied to img_res defined by data_config
    lr: 0.002
    ttur_ratio: 1.0  # controls ratio between G and D lr, where g_lr = lr and d_lr = lr * ttur_ratio
    voxel_res: 32
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



