fast-reid/projects/FastRetri/configs/cub.yml

35 lines
457 B
YAML
Raw Normal View History

2021-01-18 11:36:38 +08:00
_BASE_: base-image_retri.yml
MODEL:
LOSSES:
CE:
2021-06-17 16:59:02 +08:00
EPSILON: 0.3
2021-01-18 11:36:38 +08:00
INPUT:
SIZE_TRAIN: [256,]
SIZE_TEST: [256,]
CJ:
ENABLED: True
BRIGHTNESS: 0.25
CONTRAST: 0.25
SATURATION: 0.25
HUE: 0.0
SOLVER:
MAX_EPOCH: 30
BASE_LR: 0.02
ETA_MIN_LR: 0.00002
2021-06-23 17:49:35 +08:00
NESTEROV: False
2021-01-18 11:36:38 +08:00
MOMENTUM: 0.
TEST:
RECALLS: [ 1, 2, 4, 8, 16, 32 ]
DATASETS:
NAMES: ("CUB",)
TESTS: ("CUB",)
OUTPUT_DIR: projects/FastRetri/logs/r50-base_cub