File size: 459 Bytes
a80d6bb
 
c74a070
a80d6bb
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
from src.config.default import _CN as cfg

cfg.MODEL.MATCH_COARSE.MATCH_TYPE = "dual_softmax"
cfg.MODEL.COARSE.N_SAMPLES = 8

cfg.TRAINER.CANONICAL_LR = 1e-2
cfg.TRAINER.WARMUP_STEP = 1875  # 3 epochs
cfg.TRAINER.WARMUP_RATIO = 0.1
cfg.TRAINER.MSLR_MILESTONES = [3, 6, 9, 12, 16, 20, 24, 28]

# pose estimation
cfg.TRAINER.RANSAC_PIXEL_THR = 0.5

cfg.TRAINER.OPTIMIZER = "adamw"
cfg.TRAINER.ADAMW_DECAY = 0.1
cfg.MODEL.MATCH_COARSE.TRAIN_COARSE_PERCENT = 0.3