modify config

fastreid/config/defaults.py

@@ -150,6 +150,7 @@ _C.KD.EMA.MOMENTUM = 0.999
 
 _C.MODEL.PCB = CN()
 _C.MODEL.PCB.PART_NUM = 3
+_C.MODEL.PCB.PART_DIM = 512
 _C.MODEL.PCB.EMBEDDING_DIM = 512
 
 _C.MODEL.PCB.HEAD = CN()

projects/FastShoe/configs/r101-se-ibn.yaml

@@ -52,10 +52,10 @@ INPUT:
 
 DATALOADER:
   SAMPLER_TRAIN: TrainingSampler
-  NUM_WORKERS: 8
+  NUM_WORKERS: 16
 
 SOLVER:
-  MAX_EPOCH: 1000
+  MAX_EPOCH: 300
   AMP:
     ENABLED: True
 
@@ -63,13 +63,13 @@ SOLVER:
   SCHED: CosineAnnealingLR
 
   BASE_LR: 0.001
-  MOMENTUM: 0.9
+  MOMENTUM: 0.99
   NESTEROV: False
 
   BIAS_LR_FACTOR: 1.
   WEIGHT_DECAY: 0.0005
   WEIGHT_DECAY_BIAS: 0.
-  IMS_PER_BATCH: 16
+  IMS_PER_BATCH: 4
 
   ETA_MIN_LR: 0.00003
 
@@ -80,7 +80,7 @@ SOLVER:
 
 TEST:
   EVAL_PERIOD: 1
-  IMS_PER_BATCH: 32
+  IMS_PER_BATCH: 128
 
 DATASETS:
   NAMES: ("ShoeDataset",)

projects/FastShoe/configs/r18.yaml

@@ -0,0 +1,70 @@
+_BASE_: r101-se-ibn.yaml
+
+MODEL:
+  META_ARCHITECTURE: PCB
+
+  PCB:
+    EMBEDDING_DIM: 2048
+
+  BACKBONE:
+    NAME: build_resnet_backbone
+    DEPTH: 50x
+    NORM: BN
+    LAST_STRIDE: 2
+    FEAT_DIM: 512
+    PRETRAIN: True
+    WITH_IBN: False
+    WITH_SE: False
+
+  HEADS:
+    NAME: PcbHead
+    WITH_BNNECK: False
+    EMBEDDING_DIM: 128
+    POOL_LAYER: FastGlobalAvgPool
+    CLS_LAYER: Linear
+    NUM_CLASSES: 2
+
+  LOSSES:
+    NAME: ("CrossEntropyLoss",)
+    
+    CONTRASTIVE:
+      MARGIN: 2.0
+      SCALE: 1.
+
+DATALOADER:
+  SAMPLER_TRAIN: TrainingSampler
+  NUM_WORKERS: 8
+
+SOLVER:
+  MAX_EPOCH: 1000
+  AMP:
+    ENABLED: True
+
+  OPT: SGD
+  SCHED: CosineAnnealingLR
+
+  BASE_LR: 0.001
+  MOMENTUM: 0.9
+  NESTEROV: False
+
+  BIAS_LR_FACTOR: 1.
+  WEIGHT_DECAY: 0.0005
+  WEIGHT_DECAY_BIAS: 0.
+  IMS_PER_BATCH: 28
+
+  ETA_MIN_LR: 0.00003
+
+  WARMUP_FACTOR: 0.1
+  WARMUP_ITERS: 100
+
+  CHECKPOINT_PERIOD: 1
+
+TEST:
+  EVAL_PERIOD: 1
+  IMS_PER_BATCH: 2
+
+DATASETS:
+  NAMES: ("ShoeDataset",)
+  TESTS: ("OnlineDataset",)
+
+OUTPUT_DIR: projects/FastShoe/logs/r101_ibn_se