2021-01-18 11:36:38 +08:00
|
|
|
_BASE_: ../Base-bagtricks.yml
|
2020-05-29 16:06:04 +08:00
|
|
|
|
|
|
|
|
INPUT:
|
|
|
|
|
SIZE_TRAIN: [256, 256]
|
|
|
|
|
SIZE_TEST: [256, 256]
|
|
|
|
|
|
|
|
|
|
MODEL:
|
|
|
|
|
BACKBONE:
|
|
|
|
|
WITH_IBN: True
|
2021-02-18 11:33:05 +08:00
|
|
|
|
2020-05-29 16:06:04 +08:00
|
|
|
HEADS:
|
2021-03-31 17:07:19 +08:00
|
|
|
POOL_LAYER: GeneralizedMeanPooling
|
2021-02-18 11:33:05 +08:00
|
|
|
|
2020-05-29 16:06:04 +08:00
|
|
|
LOSSES:
|
|
|
|
|
TRI:
|
|
|
|
|
HARD_MINING: False
|
|
|
|
|
MARGIN: 0.0
|
|
|
|
|
|
|
|
|
|
DATASETS:
|
|
|
|
|
NAMES: ("VeRiWild",)
|
|
|
|
|
TESTS: ("SmallVeRiWild", "MediumVeRiWild", "LargeVeRiWild",)
|
|
|
|
|
|
|
|
|
|
SOLVER:
|
2021-03-03 12:07:39 +08:00
|
|
|
IMS_PER_BATCH: 512 # 512 For 4 GPUs
|
|
|
|
|
MAX_EPOCH: 120
|
|
|
|
|
STEPS: [30, 70, 90]
|
|
|
|
|
WARMUP_ITERS: 5000
|
2020-05-29 16:06:04 +08:00
|
|
|
|
2020-07-09 12:10:38 +08:00
|
|
|
CHECKPOINT_PERIOD: 20
|
2020-05-29 16:06:04 +08:00
|
|
|
|
|
|
|
|
TEST:
|
2021-03-03 12:07:39 +08:00
|
|
|
EVAL_PERIOD: 10
|
2020-07-09 12:10:38 +08:00
|
|
|
IMS_PER_BATCH: 128
|
2020-05-29 16:06:04 +08:00
|
|
|
|
2021-01-18 11:36:38 +08:00
|
|
|
OUTPUT_DIR: logs/veriwild/bagtricks_R50-ibn_4gpu
|
