File size: 945 Bytes
6979012 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 | 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"
|