Source: configs/synthetic_er_v5_context_subset_geometry_smoke.yaml · Plain text

configs/synthetic_er_v5_context_subset_geometry_smoke.yaml

seed: 17
device: cpu
dataset:
  name: synthetic
  root: null
  image_size: [8, 8]
  context_views: 4
  target_views: 1
  length: 2
model:
  mode: fixed
  state_architecture: evidence_residual
  voxel_resolution: [8, 8, 8]
  image_feature_dim: 8
  state_feature_dim: 8
  refinement_blocks: 1
  evidence_temperature: 0.1
  observed_residual_floor: 0.1
  completion_residual_scale: 4.0
  appearance_resolution_scale: 2
  n_samples: 8
  ray_chunk_size: 128
training:
  output_dir: outputs/synthetic_er_v5_context_subset_geometry_smoke
  batch_size: 1
  learning_rate: 0.001
  max_steps: 2
  num_workers: 0
  amp: false
  amp_dtype: bfloat16
  log_every: 1
  checkpoint_every: 2
  measurements: [rgb, depth, normal, point, visibility]
  lr_schedule: warmup_cosine
  warmup_steps: 2
  min_lr_ratio: 0.1
  ema_decay: 0.98
  evidence_residual_weight: 0.001
  context_subset_geometry_weight: 0.1