pytorch3d/projects/nerf/configs/lego.yaml

seed: 3
resume: True
stats_print_interval: 10
validation_epoch_interval: 30
checkpoint_epoch_interval: 30
checkpoint_path: 'checkpoints/lego_pt3d.pth'
data:
  dataset_name: 'lego'
  image_size: [800, 800] # [height, width]
  precache_rays: True
test:
  mode: 'evaluation'
  trajectory_type: 'circular'
  up: [0.0, 0.0, 1.0]
  scene_center: [0.0, 0.0, 0.0]
  n_frames: 100
  fps: 20
  trajectory_scale: 0.2
optimizer:
  max_epochs: 20000
  lr: 0.0005
  lr_scheduler_step_size: 5000
  lr_scheduler_gamma: 0.1
visualization:
  history_size: 10
  visdom: True
  visdom_server: 'localhost'
  visdom_port: 8097
  visdom_env: 'nerf_pytorch3d'
raysampler:
  n_pts_per_ray: 64
  n_pts_per_ray_fine: 64
  n_rays_per_image: 1024
  min_depth: 2.0
  max_depth: 6.0
  stratified: True
  stratified_test: False
  chunk_size_test: 6000
implicit_function:
  n_harmonic_functions_xyz: 10
  n_harmonic_functions_dir: 4
  n_hidden_neurons_xyz: 256
  n_hidden_neurons_dir: 128
  density_noise_std: 0.0
  n_layers_xyz: 8