You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
17 lines
457 B
17 lines
457 B
3 years ago
|
from src.config.default import _CN as cfg
|
||
|
|
||
|
cfg.LOFTR.MATCH_COARSE.MATCH_TYPE = 'sinkhorn'
|
||
|
cfg.LOFTR.MATCH_COARSE.SPARSE_SPVS = False
|
||
|
|
||
|
cfg.TRAINER.CANONICAL_LR = 8e-3
|
||
|
cfg.TRAINER.WARMUP_STEP = 1875 # 3 epochs
|
||
|
cfg.TRAINER.WARMUP_RATIO = 0.1
|
||
|
cfg.TRAINER.MSLR_MILESTONES = [8, 12, 16, 20, 24]
|
||
|
|
||
|
# pose estimation
|
||
|
cfg.TRAINER.RANSAC_PIXEL_THR = 0.5
|
||
|
|
||
|
cfg.TRAINER.OPTIMIZER = "adamw"
|
||
|
cfg.TRAINER.ADAMW_DECAY = 0.1
|
||
|
cfg.LOFTR.MATCH_COARSE.TRAIN_COARSE_PERCENT = 0.3
|