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]