Source: configs/hypersim_er_v5_phase0_pilot_train_v2.yaml · Plain text

configs/hypersim_er_v5_phase0_pilot_train_v2.yaml

seed: 17
device: auto
dataset:
  name: hypersim
  root: data/hypersim_er_prepared/train
  image_size:
  - 128
  - 160
  context_views: 4
  target_views: 2
  length: 16
  sample_stride: 2
  spatial_protocol: context_local_metric
  local_bounds_m:
  - - -6.0
    - -4.0
    - 0.05
  - - 6.0
    - 4.0
    - 12.05
  scene_ids:
  - ai_037_003
  - ai_031_007
  - ai_048_003
  - ai_038_010
  - ai_003_006
  - ai_004_001
  - ai_047_002
  - ai_031_006
  - ai_018_010
  - ai_039_010
  - ai_009_006
  - ai_015_007
  - ai_046_001
  - ai_009_003
  - ai_026_001
  - ai_029_003
model:
  mode: fixed
  state_architecture: evidence_residual
  voxel_resolution:
  - 48
  - 32
  - 48
  image_feature_dim: 48
  state_feature_dim: 48
  refinement_blocks: 4
  evidence_temperature: 0.1
  observed_residual_floor: 0.1
  completion_residual_scale: 4.0
  appearance_resolution_scale: 2
  n_samples: 64
  ray_chunk_size: 4096
training:
  output_dir: outputs\hypersim_er_v6_phase0_v5_pilot_train_v2
  batch_size: 1
  learning_rate: 0.0002
  max_steps: 10000
  num_workers: 0
  amp: true
  amp_dtype: bfloat16
  log_every: 20
  checkpoint_every: 5000
  measurements:
  - rgb
  - depth
  - normal
  - point
  - visibility
  gradient_accumulation: 2
  weight_decay: 0.0001
  lr_schedule: warmup_cosine
  warmup_steps: 3000
  min_lr_ratio: 0.05
  ema_decay: 0.98
  evidence_residual_weight: 0.001