| 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" | |