mirrors/fast-reid
1
0
Fork 0
mirror of https://github.com/JDAI-CV/fast-reid.git synced 2025-06-03 14:50:47 +08:00
Code Issues Projects Releases Wiki Activity

change config file

Browse Source
This commit is contained in:
liaoxingyu 2020-09-10 10:58:56 +08:00
parent 4d573b8107
commit 3c48bf78c1
3 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
configs/Base-MGN.yml
View File

@ -9,9 +9,7 @@ MODEL:
WITH_NL: False
HEADS:
NAME: "ReductionHead"
POOL_LAYER: "maxpool"
REDUCTION_DIM: 256
EMBEDDING_DIM: 256
LOSSES:
NAME: ("CrossEntropyLoss", "TripletLoss",)

2
configs/Base-Strongerbaseline.yml
View File

@ -8,7 +8,7 @@ MODEL:
HEADS:
NECK_FEAT: "after"
POOL_LAYER: "gempool"
POOL_LAYER: "gempoolP"
CLS_LAYER: "circleSoftmax"
SCALE: 64
MARGIN: 0.35

5
configs/Base-bagtricks.yml
View File

@ -6,12 +6,15 @@ MODEL:
NORM: "BN"
DEPTH: "50x"
LAST_STRIDE: 1
FEAT_DIM: 2048
WITH_IBN: False
PRETRAIN: True
PRETRAIN_PATH: "/export/home/lxy/.cache/torch/checkpoints/resnet50-19c8e357.pth"
HEADS:
NAME: "BNneckHead"
NAME: "EmbeddingHead"
NORM: "BN"
WITH_BNNECK: True
POOL_LAYER: "avgpool"
NECK_FEAT: "before"
CLS_LAYER: "linear"