seed: 42

model:
  num_keypoints: 512

  prune:
    backbone_0_1: 16
    backbone_1_0: 16
    backbone_1_1: 24
    backbone_2_0: 32
    backbone_2_1: 64
  
data:
  train_image_dir: "data/datasets/indoor_forward_3_snapdragon_with_gt/img"
  train_image_ids_file: "data/gt_id_file_train_base_250.txt"
  val_image_dir: "data/datasets/indoor_forward_3_snapdragon_with_gt/img"
  val_image_ids_file: "data/gt_id_file_train_base_250.txt"
  ground_truth_dir: "data"
  keypoints_file: "kpts_train_base_250.npy"
  descriptors_file: "desc_train_base_250.npy"
  array_ids_file: "gt_id_file_train_base_250.txt"
  image_size: [640, 480]
  batch_size: 16
  num_workers: 2
  pin_memory: true

optimizer:
  lr: 1.0e-3
  weight_decay: 1.0e-3

loss:
  cross_entropy_coef: 2.0
  kl_coef: 2.0
  descriptor_coef: 1.0

trainer:
  max_epochs: 300
  accelerator: "auto"
  devices: 1
  precision: "32-true"
  log_every_n_steps: 5
  default_root_dir: "./distillation_runs"