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

minor fix for configs 

#519
pull/525/head
liaoxingyu 2021-06-17 16:59:02 +08:00
parent 2fee8327fa
commit 7c1269d6c3
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
projects/FastRetri/configs/base-image_retri.yml
View File

@ -12,6 +12,7 @@ MODEL:
HEADS: HEADS:
NAME: EmbeddingHead NAME: EmbeddingHead
NORM: syncBN NORM: syncBN
WITH_BNNECK: True
NECK_FEAT: after NECK_FEAT: after
EMBEDDING_DIM: 0 EMBEDDING_DIM: 0
POOL_LAYER: GeneralizedMeanPooling POOL_LAYER: GeneralizedMeanPooling
@ -52,7 +53,7 @@ DATALOADER:
SOLVER: SOLVER:
MAX_EPOCH: 100 MAX_EPOCH: 100
AMP: AMP:
ENABLED: False ENABLED: True
OPT: SGD OPT: SGD
SCHED: CosineAnnealingLR SCHED: CosineAnnealingLR
@ -73,6 +74,9 @@ SOLVER:
CHECKPOINT_PERIOD: 10 CHECKPOINT_PERIOD: 10
CLIP_GRADIENTS:
ENABLED: True
TEST: TEST:
EVAL_PERIOD: 10 EVAL_PERIOD: 10
IMS_PER_BATCH: 256 IMS_PER_BATCH: 256

2
projects/FastRetri/configs/cub.yml
View File

@ -3,7 +3,7 @@ _BASE_: base-image_retri.yml
MODEL: MODEL:
LOSSES: LOSSES:
CE: CE:
EPSILON: 0.4 EPSILON: 0.3
INPUT: INPUT:
SIZE_TRAIN: [256,] SIZE_TRAIN: [256,]