# Fine-tuning on real data config
phase: finetune

dataset: fusion360
model: baseline

pretrained_checkpoint: checkpoints/pretrain/best_val_loss.ckpt

optimizer:
  name: adamw
  lr: 1e-5
  weight_decay: 1e-4

scheduler:
  name: cosine_annealing
  T_max: 50
  eta_min: 1e-7

training:
  epochs: 50
  batch_size: 32
  gradient_clip: 1.0
  early_stopping_patience: 10
  amp: true
  freeze_encoder: false  # set true for frozen encoder experiment

loss:
  edge_weight: 1.0
  graph_weight: 0.5
  joint_type_weight: 0.3
  dof_weight: 0.2
  redundant_penalty: 2.0

checkpointing:
  save_best_val_loss: true
  save_best_val_accuracy: true
  save_every_n_epochs: 5
  checkpoint_dir: checkpoints/finetune

logging:
  backend: wandb
  project: kindred-solver
  log_every_n_steps: 20

seed: 42
