Source: configs/replica_pilot.yaml · Plain text

configs/replica_pilot.yaml

seed: 17
device: auto
dataset:
  name: replica
  root: data/replica_prepared
  image_size: [128, 160]
  context_views: 2
  target_views: 1
  length: 1
  sample_stride: 5
model:
  mode: fixed
  voxel_resolution: 32
  image_feature_dim: 32
  state_feature_dim: 32
  refinement_blocks: 2
  n_samples: 48
  ray_chunk_size: 4096
training:
  output_dir: outputs/replica_pilot
  batch_size: 1
  learning_rate: 0.0005
  max_steps: 2000
  num_workers: 0
  amp: true
  log_every: 20
  checkpoint_every: 250
  measurements: [rgb, depth, visibility]