mirror of
https://github.com/JDAI-CV/fast-reid.git
synced 2025-06-02 22:29:29 +08:00
37 lines
661 B
YAML
37 lines
661 B
YAML
_BASE_: "../Base-bagtricks.yml"
|
|
|
|
INPUT:
|
|
SIZE_TRAIN: [256, 256]
|
|
SIZE_TEST: [256, 256]
|
|
|
|
MODEL:
|
|
BACKBONE:
|
|
WITH_IBN: True
|
|
PRETRAIN_PATH: '/export2/home/zjk/pretrain_models/resnet50_ibn_a.pth.tar'
|
|
HEADS:
|
|
NUM_CLASSES: 30671
|
|
POOL_LAYER: gempool
|
|
LOSSES:
|
|
TRI:
|
|
HARD_MINING: False
|
|
MARGIN: 0.0
|
|
|
|
DATASETS:
|
|
NAMES: ("VeRiWild",)
|
|
TESTS: ("SmallVeRiWild", "MediumVeRiWild", "LargeVeRiWild",)
|
|
|
|
SOLVER:
|
|
IMS_PER_BATCH: 512
|
|
MAX_ITER: 64800
|
|
STEPS: [21600, 48600]
|
|
WARMUP_ITERS: 5400
|
|
|
|
LOG_PERIOD: 540
|
|
CHECKPOINT_PERIOD: 21600
|
|
|
|
TEST:
|
|
EVAL_PERIOD: 5400
|
|
IMS_PER_BATCH: 1024
|
|
|
|
OUTPUT_DIR: "logs/veriwild/bagtricks_R50-ibn_4gpu"
|