debug: false
seed: 42
enable_xformers: null

training:
  num_train_epochs: 10
  max_train_steps: null
  gradient_accumulation_steps: 1
  only_train_attention_layers: true
  gradient_checkpointing: true
  resume_path: null
  use_EMA: true

model:
  modality: normals
  pretrained_path: GonzaloMG/marigold-e2e-ft-depth
  checkpoint_path: null
  unet_positional_encoding: RoPE
  vae_use_RoPE: true
  metric_depth: null
  log_scale: null

data:
  data_path: null
  dataset: PanoInfinigen
  scenes: both
  batch_size: 1
  use_data_augmentation: true

optimization:
  learning_rate: 3e-4
  lr_exp_warmup_steps: 0.03
  adam_beta1: 0.9
  adam_beta2: 0.999
  adam_weight_decay: 1e-2
  adam_epsilon: 1e-8
  clip_grad_norm: true
  max_grad_norm: 1.0

loss:
  l1_loss_weight: null
  grad_loss_weight: null
  normals_consistency_loss_weight: null
  invalid_mask_weight: null

validation:
  run_validation: true
  run_tiny_validation: true
  tiny_val_frequency: 200

logging:
  tracker_project_name: Pager
  model_save_path: checkpoints/
  model_save_frequency: 1
  loss_report_frequency: 1
  img_report_frequency: 200
  report_to: wandb

  tensorboard:
    log_path: logs

  wandb:
    run_name: PaGeR-normals