mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

update online-pcb.yaml

pull/608/head
zuchen.wang 2021-11-12 19:39:40 +08:00
parent 781b989407
commit bc63002d05
1 changed files with 21 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
projects/FastShoe/configs/online-pcb.yaml
View File

@ -11,17 +11,15 @@ MODEL:
HEAD: HEAD:
FULL_DIM: 2048 FULL_DIM: 2048
PART_DIM: 512 PART_DIM: 512
EMBEDDING_DIM: 512 EMBEDDING_DIM: 128
BACKBONE: BACKBONE:
PRETRAIN: True PRETRAIN: False
PRETRAIN_PATH: /home/apps/.cache/torch/hub/checkpoints/se_resnext101_32x4d-3b2fe3d8.pth
NAME: build_senet_pcb_backbone NAME: build_senet_pcb_backbone
DEPTH: 101x DEPTH: 101x
NORM: BN NORM: BN
LAST_STRIDE: 1 LAST_STRIDE: 1
FEAT_DIM: 512 FEAT_DIM: 512
PRETRAIN: True
WITH_IBN: True WITH_IBN: True
WITH_SE: True WITH_SE: True
@ -34,15 +32,15 @@ MODEL:
NUM_CLASSES: 2 NUM_CLASSES: 2
LOSSES: LOSSES:
NAME: ("CrossEntropyLoss",) NAME: ("BinaryFocalLoss",)
BFL:
CONTRASTIVE: ALPHA: 0.25
MARGIN: 2.0 GAMMA: 2
SCALE: 1. SCALE: 1.0
INPUT: INPUT:
SIZE_TRAIN: [310, 300] SIZE_TRAIN: [310, 300]
SIZE_TEST: [310, 300] SIZE_TEST: [270, 260]
CROP: CROP:
ENABLED: True ENABLED: True
@ -54,12 +52,15 @@ DATALOADER:
NUM_WORKERS: 8 NUM_WORKERS: 8
SOLVER: SOLVER:
AMP:
ENABLED: True
OPT: SGD OPT: SGD
SCHED: CosineAnnealingLR SCHED: MultiStepLR
BASE_LR: 0.001 BASE_LR: 0.001
MOMENTUM: 0.9 MOMENTUM: 0.9
NESTEROV: False NESTEROV: True
BIAS_LR_FACTOR: 1. BIAS_LR_FACTOR: 1.
WEIGHT_DECAY: 0.0005 WEIGHT_DECAY: 0.0005
@ -67,16 +68,17 @@ SOLVER:
ETA_MIN_LR: 0.00003 ETA_MIN_LR: 0.00003
WARMUP_FACTOR: 0.1 WARMUP_FACTOR: 0.1
WARMUP_ITERS: 1000 WARMUP_ITERS: 0
MAX_EPOCH: 200 MAX_EPOCH: 100
IMS_PER_BATCH: 150 IMS_PER_BATCH: 80
TEST: TEST:
IMS_PER_BATCH: 512 IMS_PER_BATCH: 480
DATASETS: DATASETS:
NAMES: ("PairDataset",) NAMES: ("PairDataset",)
TESTS: ("PairDataset", "ExcelDataset") TESTS: ("PairDataset", "ExcelDataset",)
OUTPUT_DIR: projects/FastShoe/logs/online-pcb OUTPUT_DIR: projects/FastShoe/logs/online-pcb