chore(configs): update all training config

MODEL_ZOO.md

@@ -9,8 +9,8 @@
 
 | Method | Pretrained | Rank@1 | mAP | mINP |
 | :---: | :---: | :---: |:---: | :---: |
-| BoT(R50) | ImageNet | 93.6% | 85.1% | 58.1% |
+| BoT(R50) | ImageNet | 94.1% | 85.9% | 59.3% |
-| BoT(R50-ibn) | ImageNet | 94.8% | 87.3% | 63.5% |
+| BoT(R50-ibn) | ImageNet | - | - | - |
 | BoT(S50) | ImageNet | - | - | - |
 | BoT(R101-ibn) | ImageNet| - | - |
 
@@ -27,7 +27,7 @@
 | Method | Pretrained | Rank@1 | mAP | mINP |
 | :---: | :---: | :---: |:---: | :---: |
 | SBS(R50) | ImageNet | - | - | - |
-| SBS(R50-ibn) | ImageNet | 95.5% | 88.4% | 65.8% |
+| SBS(R50-ibn) | ImageNet | 95.5% | 88.7% | 66.4% |
 | SBS(S50) | ImageNet | - | - | - |
 | SBS(R101-ibn) | ImageNet | - | - | - |
 

configs/Base-AGW.yml

@@ -11,6 +11,7 @@ MODEL:
   HEADS:
     NAME: "BNneckHead"
     POOL_LAYER: "gempool"
+    NECK_FEAT: "before"
     CLS_LAYER: "linear"
     NUM_CLASSES: 702
 
@@ -46,14 +47,14 @@ DATALOADER:
 
 SOLVER:
   OPT: "Adam"
-  MAX_ITER: 18000
+  MAX_ITER: 24000
   BASE_LR: 0.00035
   BIAS_LR_FACTOR: 2.
   WEIGHT_DECAY: 0.0005
   WEIGHT_DECAY_BIAS: 0.0005
   IMS_PER_BATCH: 64
 
-  STEPS: [8000, 14000]
+  STEPS: [8000, 18000]
   GAMMA: 0.1
 
   WARMUP_FACTOR: 0.01

configs/Base-Strongerbaseline.yml

@@ -5,6 +5,7 @@ MODEL:
 
   BACKBONE:
     NAME: "build_resnet_backbone"
+    NORM: "BN"
     DEPTH: 50
     LAST_STRIDE: 1
     WITH_IBN: False
@@ -13,10 +14,12 @@ MODEL:
 
   HEADS:
     NAME: "BNneckHead"
+    NORM: "BN"
+    NECK_FEAT: "after"
     CLS_LAYER: "circle"
     POOL_LAYER: "gempool"
-    SCALE: 128
+    SCALE: 64
-    MARGIN: 0.25
+    MARGIN: 0.35
 
   LOSSES:
     NAME: ("CrossEntropyLoss", "TripletLoss",)
@@ -61,12 +64,9 @@ SOLVER:
 
   SCHED: "DelayedCosineAnnealingLR"
 
-  DELAY_ITERS: 2000
+  DELAY_ITERS: 9000
   ETA_MIN_LR: 0.00000077
 
-  STEPS: [8000, 14000]
-  GAMMA: 0.1
-
   WARMUP_FACTOR: 0.01
   WARMUP_ITERS: 2000
 
@@ -79,7 +79,7 @@ TEST:
   EVAL_PERIOD: 2000
   IMS_PER_BATCH: 512
   PRECISE_BN:
-    ENABLED: True
+    ENABLED: False
     DATASET: 'DukeMTMC'
 
 CUDNN_BENCHMARK: True

configs/Base-bagtricks.yml

@@ -1,10 +1,11 @@
 MODEL:
-  META_ARCHITECTURE: 'Baseline'
+  META_ARCHITECTURE: "Baseline"
 
   OPEN_LAYERS: ""
 
   BACKBONE:
     NAME: "build_resnet_backbone"
+    NORM: "BN"
     DEPTH: 50
     LAST_STRIDE: 1
     WITH_IBN: False
@@ -12,20 +13,22 @@ MODEL:
 
   HEADS:
     NAME: "BNneckHead"
+    NORM: "BN"
+    NECK_FEAT: "before"
     CLS_LAYER: "linear"
 
   LOSSES:
-    NAME: ("CrossEntropyLoss", "TripletLoss")
+    NAME: ("CrossEntropyLoss", "TripletLoss", )
 
     CE:
       EPSILON: 0.1
-      SCALE: 1.0
+      SCALE: 1.
 
     TRI:
       MARGIN: 0.3
       HARD_MINING: True
       USE_COSINE_DIST: False
-      SCALE: 1.0
+      SCALE: 1.
 
 DATASETS:
   NAMES: ("DukeMTMC",)
@@ -48,14 +51,14 @@ DATALOADER:
 
 SOLVER:
   OPT: "Adam"
-  MAX_ITER: 18000
+  MAX_ITER: 24000
   BASE_LR: 0.00035
   BIAS_LR_FACTOR: 2.
   WEIGHT_DECAY: 0.0005
-  WEIGHT_DECAY_BIAS: 0.
+  WEIGHT_DECAY_BIAS: 0.0005
   IMS_PER_BATCH: 64
 
-  STEPS: [8000, 14000]
+  STEPS: [8000, 18000]
   GAMMA: 0.1
 
   WARMUP_FACTOR: 0.01

configs/DukeMTMC/AGW.yml

@@ -1,18 +0,0 @@
-_BASE_: "../Base-AGW_R50.yml"
-
-MODEL:
-  HEADS:
-    NUM_CLASSES: 702
-
-SOLVER:
-  MAX_ITER: 23000
-
-  STEPS: [10000, 18000]
-
-  WARMUP_ITERS: 2500
-
-DATASETS:
-  NAMES: ("DukeMTMC",)
-  TESTS: ("DukeMTMC",)
-
-OUTPUT_DIR: "logs/dukemtmc/agw"

configs/DukeMTMC/AGW_R50-ibn.yml

@@ -0,0 +1,16 @@
+_BASE_: "../Base-AGW.yml"
+
+MODEL:
+  BACKBONE:
+    NAME: "build_resnet_backbone"
+    WITH_IBN: True
+    PRETRAIN_PATH: "/home/liaoxingyu2/lxy/.cache/torch/checkpoints/resnet50_ibn_a.pth.tar"
+
+  HEADS:
+    NUM_CLASSES: 702
+
+DATASETS:
+  NAMES: ("DukeMTMC",)
+  TESTS: ("DukeMTMC",)
+
+OUTPUT_DIR: "logs/dukemtmc/agw_R50-ibn"

configs/DukeMTMC/AGW_R50.yml

@@ -0,0 +1,12 @@
+_BASE_: "../Base-AGW.yml"
+
+MODEL:
+  HEADS:
+    NUM_CLASSES: 702
+
+
+DATASETS:
+  NAMES: ("DukeMTMC",)
+  TESTS: ("DukeMTMC",)
+
+OUTPUT_DIR: "logs/dukemtmc/agw_R50"

configs/DukeMTMC/AGW_S50.yml

@@ -0,0 +1,15 @@
+_BASE_: "../Base-AGW.yml"
+
+MODEL:
+  BACKBONE:
+    NAME: "build_resnest_backbone"
+
+  HEADS:
+    NECK_FEAT: "after"
+    NUM_CLASSES: 702
+
+DATASETS:
+  NAMES: ("DukeMTMC",)
+  TESTS: ("DukeMTMC",)
+
+OUTPUT_DIR: "logs/dukemtmc/agw_S50"

configs/DukeMTMC/bagtricks_R101-ibn.yml

@@ -9,16 +9,8 @@ MODEL:
   HEADS:
     NUM_CLASSES: 702
 
-SOLVER:
-  MAX_ITER: 18000
-
-  STEPS: [8000, 14000]
-
-  WARMUP_ITERS: 2000
-
 DATASETS:
   NAMES: ("DukeMTMC",)
   TESTS: ("DukeMTMC",)
 
 OUTPUT_DIR: "logs/dukemtmc/bagtricks_R101-ibn"
-

configs/DukeMTMC/bagtricks_R50-ibn.yml

@@ -8,16 +8,8 @@ MODEL:
   HEADS:
     NUM_CLASSES: 702
 
-SOLVER:
-  MAX_ITER: 18000
-
-  STEPS: [8000, 14000]
-
-  WARMUP_ITERS: 2000
-
 DATASETS:
   NAMES: ("DukeMTMC",)
   TESTS: ("DukeMTMC",)
 
 OUTPUT_DIR: "logs/dukemtmc/bagtricks_R50-ibn"
-

configs/DukeMTMC/bagtricks_R50.yml

@@ -4,12 +4,6 @@ MODEL:
   HEADS:
     NUM_CLASSES: 702
 
-SOLVER:
-  MAX_ITER: 18000
-
-  STEPS: [8000, 14000]
-
-  WARMUP_ITERS: 2000
 
 DATASETS:
   NAMES: ("DukeMTMC",)

configs/DukeMTMC/bagtricks_S50.yml

@@ -7,16 +7,8 @@ MODEL:
   HEADS:
     NUM_CLASSES: 702
 
-SOLVER:
-  MAX_ITER: 18000
-
-  STEPS: [8000, 14000]
-
-  WARMUP_ITERS: 2000
-
 DATASETS:
   NAMES: ("DukeMTMC",)
   TESTS: ("DukeMTMC",)
 
 OUTPUT_DIR: "logs/dukemtmc/bagtricks_S50"
-

configs/DukeMTMC/sbs.yml

@@ -1,23 +0,0 @@
-_BASE_: "../Base-Strongerbaseline.yml"
-
-MODEL:
-  BACKBONE:
-    NAME: "build_resnest_backbone"
-    WITH_IBN: False
-    WITH_NL: True
-    PRETRAIN_PATH: "/home/liaoxingyu2/lxy/.cache/torch/checkpoints/resnet50_ibn_a.pth.tar"
-
-  HEADS:
-    NUM_CLASSES: 702
-
-SOLVER:
-  MAX_ITER: 5
-  DELAY_ITERS: 8000
-  WARMUP_ITERS: 2600
-  FREEZE_ITERS: 2600
-
-DATASETS:
-  NAMES: ("DukeMTMC",)
-  TESTS: ("DukeMTMC",)
-
-OUTPUT_DIR: "logs/dukemtmc/sbs"

configs/DukeMTMC/sbs_R101-ibn.yml

@@ -0,0 +1,21 @@
+_BASE_: "../Base-Strongerbaseline.yml"
+
+MODEL:
+  BACKBONE:
+    DEPTH: 101
+    WITH_NL: False
+    WITH_IBN: True
+    PRETRAIN_PATH: "/export/home/lxy/.cache/torch/checkpoints/resnet101_ibn_a.pth.tar"
+
+  HEADS:
+    NUM_CLASSES: 702
+
+DATASETS:
+  NAMES: ("DukeMTMC",)
+  TESTS: ("DukeMTMC",)
+
+TEST:
+  PRECISE_BN:
+    DATASET: 'DukeMTMC'
+
+OUTPUT_DIR: "logs/dukemtmc/sbs-R101-ibn"

configs/DukeMTMC/sbs_R50-ibn.yml

@@ -0,0 +1,20 @@
+_BASE_: "../Base-Strongerbaseline.yml"
+
+MODEL:
+  BACKBONE:
+    WITH_IBN: True
+    PRETRAIN_PATH: "/export/home/lxy/.cache/torch/checkpoints/resnet50_ibn_a.pth.tar"
+
+  HEADS:
+    NUM_CLASSES: 702
+
+DATASETS:
+  NAMES: ("DukeMTMC",)
+  TESTS: ("DukeMTMC",)
+
+TEST:
+  EVAL_PERIOD: 2000
+  PRECISE_BN:
+    DATASET: 'DukeMTMC'
+
+OUTPUT_DIR: "logs/dukemtmc/sbs_R50-ibn"

configs/DukeMTMC/sbs_R50.yml

@@ -0,0 +1,15 @@
+_BASE_: "../Base-Strongerbaseline.yml"
+
+MODEL:
+  HEADS:
+    NUM_CLASSES: 702
+
+DATASETS:
+  NAMES: ("DukeMTMC",)
+  TESTS: ("DukeMTMC",)
+
+TEST:
+  PRECISE_BN:
+    DATASET: 'DukeMTMC'
+
+OUTPUT_DIR: "logs/dukemtmc/sbs_R50"

configs/DukeMTMC/sbs_S50.yml

@@ -0,0 +1,22 @@
+_BASE_: "../Base-Strongerbaseline.yml"
+
+MODEL:
+  BACKBONE:
+    NAME: "build_resnest_backbone"
+
+  HEADS:
+    NUM_CLASSES: 702
+
+  LOSSES:
+    TRI:
+      MARGIN: 0.3
+
+DATASETS:
+  NAMES: ("DukeMTMC",)
+  TESTS: ("DukeMTMC",)
+
+TEST:
+  PRECISE_BN:
+    DATASET: 'DukeMTMC'
+
+OUTPUT_DIR: "logs/dukemtmc/sbs_S50"

configs/MSMT17/AGW_R50-ibn.yml

@@ -0,0 +1,25 @@
+_BASE_: "../Base-AGW.yml"
+
+MODEL:
+  BACKBONE:
+    WITH_IBN: True
+    PRETRAIN_PATH: "/export/home/lxy/.cache/torch/checkpoints/resnet50_ibn_a.pth.tar"
+
+  HEADS:
+    NUM_CLASSES: 1041
+
+DATASETS:
+  NAMES: ("MSMT17",)
+  TESTS: ("MSMT17",)
+
+SOLVER:
+  MAX_ITER: 42000
+  STEPS: [19000, 33000]
+  WARMUP_ITERS: 4700
+
+  CHECKPOINT_PERIOD: 5000
+
+TEST:
+  EVAL_PERIOD: 5000
+
+OUTPUT_DIR: "logs/msmt17/agw_R50-ibn"

configs/MSMT17/AGW_R50.yml

@@ -1,4 +1,4 @@
-_BASE_: "../Base-AGW_R50.yml"
+_BASE_: "../Base-AGW.yml"
 
 MODEL:
   HEADS:
@@ -18,5 +18,4 @@ SOLVER:
 TEST:
   EVAL_PERIOD: 5000
 
-OUTPUT_DIR: "logs/msmt17/agw"
+OUTPUT_DIR: "logs/msmt17/agw_R50"
-

configs/MSMT17/AGW_S50.yml

@@ -0,0 +1,24 @@
+_BASE_: "../Base-AGW.yml"
+
+MODEL:
+  BACKBONE:
+    NAME: "build_resnest_backbone"
+
+  HEADS:
+    NUM_CLASSES: 1041
+
+DATASETS:
+  NAMES: ("MSMT17",)
+  TESTS: ("MSMT17",)
+
+SOLVER:
+  MAX_ITER: 42000
+  STEPS: [19000, 33000]
+  WARMUP_ITERS: 4700
+
+  CHECKPOINT_PERIOD: 5000
+
+TEST:
+  EVAL_PERIOD: 5000
+
+OUTPUT_DIR: "logs/msmt17/agw_S50"

configs/MSMT17/bagtricks_R101-ibn.yml

@@ -4,22 +4,23 @@ MODEL:
   BACKBONE:
     DEPTH: 101
     WITH_IBN: True
-    PRETRAIN_PATH: "/home/liaoxingyu2/lxy/.cache/torch/checkpoints/resnet50_ibn_a.pth.tar"
+    PRETRAIN_PATH: "/export/home/lxy/.cache/torch/checkpoints/resnet101_ibn_a.pth.tar"
 
   HEADS:
     NUM_CLASSES: 1041
 
 SOLVER:
   MAX_ITER: 42000
-
   STEPS: [19000, 33000]
-
   WARMUP_ITERS: 4700
-
+  CHECKPOINT_PERIOD: 5000
 
 DATASETS:
   NAMES: ("MSMT17",)
   TESTS: ("MSMT17",)
 
-OUTPUT_DIR: "logs/msmt/bagtricks_R101-ibn"
+TEST:
+  EVAL_PERIOD: 5000
+
+OUTPUT_DIR: "logs/msmt17/bagtricks_R101-ibn"
 

configs/MSMT17/bagtricks_R50-ibn.yml

@@ -3,22 +3,23 @@ _BASE_: "../Base-bagtricks.yml"
 MODEL:
   BACKBONE:
     WITH_IBN: True
-    PRETRAIN_PATH: "/home/liaoxingyu2/lxy/.cache/torch/checkpoints/resnet50_ibn_a.pth.tar"
+    PRETRAIN_PATH: "/export/home/lxy/.cache/torch/checkpoints/resnet50_ibn_a.pth.tar"
 
   HEADS:
     NUM_CLASSES: 1041
 
 SOLVER:
   MAX_ITER: 42000
-
   STEPS: [19000, 33000]
-
   WARMUP_ITERS: 4700
-
+  CHECKPOINT_PERIOD: 5000
 
 DATASETS:
   NAMES: ("MSMT17",)
   TESTS: ("MSMT17",)
 
-OUTPUT_DIR: "logs/msmt/bagtricks_R50-ibn"
+TEST:
+  EVAL_PERIOD: 5000
+
+OUTPUT_DIR: "logs/msmt17/bagtricks_R50-ibn"
 

configs/MSMT17/bagtricks_R50.yml

@@ -6,13 +6,15 @@ MODEL:
 
 SOLVER:
   MAX_ITER: 42000
-
   STEPS: [19000, 33000]
-
   WARMUP_ITERS: 4700
+  CHECKPOINT_PERIOD: 5000
 
 DATASETS:
   NAMES: ("MSMT17",)
   TESTS: ("MSMT17",)
 
-OUTPUT_DIR: "logs/msmt/bagtricks_R50"
+TEST:
+  EVAL_PERIOD: 5000
+
+OUTPUT_DIR: "logs/msmt17/bagtricks_R50"

configs/MSMT17/bagtricks_S50.yml

@@ -9,15 +9,16 @@ MODEL:
 
 SOLVER:
   MAX_ITER: 42000
-
   STEPS: [19000, 33000]
-
   WARMUP_ITERS: 4700
-
+  CHECKPOINT_PERIOD: 5000
 
 DATASETS:
   NAMES: ("MSMT17",)
   TESTS: ("MSMT17",)
 
-OUTPUT_DIR: "logs/msmt/bagtricks_S50"
+TEST:
+  EVAL_PERIOD: 5000
+
+OUTPUT_DIR: "logs/msmt17/bagtricks_S50"
 

configs/MSMT17/sbs.yml

@@ -1,29 +0,0 @@
-_BASE_: "../Base-Strongerbaseline.yml"
-
-MODEL:
-  BACKBONE:
-    NAME: "build_resnest_backbone"
-    WITH_IBN: False
-    WITH_NL: True
-    PRETRAIN_PATH: "/home/liaoxingyu2/lxy/.cache/torch/checkpoints/resnet50_ibn_a.pth.tar"
-
-  HEADS:
-    NUM_CLASSES: 1041
-
-DATASETS:
-  NAMES: ("MSMT17",)
-  TESTS: ("MSMT17",)
-
-
-SOLVER:
-  MAX_ITER: 29000
-  DELAY_ITERS: 14000
-  WARMUP_ITERS: 4700
-  FREEZE_ITERS: 4700
-
-  CHECKPOINT_PERIOD: 4000
-
-TEST:
-  EVAL_PERIOD: 4000
-
-OUTPUT_DIR: "logs/msmt17/resnest-nl-gem-circle_s64m0.35_loss-cos_delay-autoaug"

configs/MSMT17/sbs_R101-ibn.yml

@@ -0,0 +1,28 @@
+_BASE_: "../Base-Strongerbaseline.yml"
+
+MODEL:
+  BACKBONE:
+    DEPTH: 101
+    WITH_NL: False
+    WITH_IBN: True
+    PRETRAIN_PATH: "/export/home/lxy/.cache/torch/checkpoints/resnet101_ibn_a.pth.tar"
+
+  HEADS:
+    NUM_CLASSES: 1041
+
+DATASETS:
+  NAMES: ("MSMT17",)
+  TESTS: ("MSMT17",)
+
+SOLVER:
+  MAX_ITER: 28000
+  DELAY_ITERS: 14000
+  WARMUP_ITERS: 4700
+  FREEZE_ITERS: 4700
+
+TEST:
+  EVAL_PERIOD: 5000
+  PRECISE_BN:
+    DATASET: 'MSMT17'
+
+OUTPUT_DIR: "logs/msmt17/sbs-R101-ibn"

configs/MSMT17/sbs_R50-ibn.yml

@@ -0,0 +1,26 @@
+_BASE_: "../Base-Strongerbaseline.yml"
+
+MODEL:
+  BACKBONE:
+    WITH_IBN: True
+    PRETRAIN_PATH: "/export/home/lxy/.cache/torch/checkpoints/resnet50_ibn_a.pth.tar"
+
+  HEADS:
+    NUM_CLASSES: 1041
+
+DATASETS:
+  NAMES: ("MSMT17",)
+  TESTS: ("MSMT17",)
+
+SOLVER:
+  MAX_ITER: 28000
+  DELAY_ITERS: 14000
+  WARMUP_ITERS: 4700
+  FREEZE_ITERS: 4700
+
+TEST:
+  EVAL_PERIOD: 5000
+  PRECISE_BN:
+    DATASET: 'MSMT17'
+
+OUTPUT_DIR: "logs/msmt17/sbs_R50-ibn"

configs/MSMT17/sbs_R50.yml

@@ -0,0 +1,22 @@
+_BASE_: "../Base-Strongerbaseline.yml"
+
+MODEL:
+  HEADS:
+    NUM_CLASSES: 1041
+
+DATASETS:
+  NAMES: ("MSMT17",)
+  TESTS: ("MSMT17",)
+
+SOLVER:
+  MAX_ITER: 28000
+  DELAY_ITERS: 14000
+  WARMUP_ITERS: 4700
+  FREEZE_ITERS: 4700
+
+TEST:
+  EVAL_PERIOD: 5000
+  PRECISE_BN:
+    DATASET: 'MSMT17'
+
+OUTPUT_DIR: "logs/msmt17/sbs_R50"

configs/MSMT17/sbs_S50.yml

@@ -0,0 +1,29 @@
+_BASE_: "../Base-Strongerbaseline.yml"
+
+MODEL:
+  BACKBONE:
+    NAME: "build_resnest_backbone"
+
+  HEADS:
+    NUM_CLASSES: 1041
+
+  LOSSES:
+    TRI:
+      MARGIN: 0.3
+
+DATASETS:
+  NAMES: ("MSMT17",)
+  TESTS: ("MSMT17",)
+
+SOLVER:
+  MAX_ITER: 28000
+  DELAY_ITERS: 14000
+  WARMUP_ITERS: 4700
+  FREEZE_ITERS: 4700
+
+TEST:
+  EVAL_PERIOD: 5000
+  PRECISE_BN:
+    DATASET: 'MSMT17'
+
+OUTPUT_DIR: "logs/msmt17/sbs_S50"

configs/Market1501/AGW_R50-ibn.yml

@@ -9,12 +9,6 @@ MODEL:
   HEADS:
     NUM_CLASSES: 751
 
-SOLVER:
-  MAX_ITER: 18000
-
-  STEPS: [8000, 14000]
-
-  WARMUP_ITERS: 2000
 
 DATASETS:
   NAMES: ("Market1501",)

configs/Market1501/AGW_R50.yml

@@ -4,12 +4,6 @@ MODEL:
   HEADS:
     NUM_CLASSES: 751
 
-SOLVER:
-  MAX_ITER: 18000
-
-  STEPS: [8000, 14000]
-
-  WARMUP_ITERS: 2000
 
 DATASETS:
   NAMES: ("Market1501",)

configs/Market1501/AGW_S50.yml

@@ -5,15 +5,9 @@ MODEL:
     NAME: "build_resnest_backbone"
 
   HEADS:
+    NECK_FEAT: "after"
     NUM_CLASSES: 751
 
-SOLVER:
-  MAX_ITER: 18000
-
-  STEPS: [8000, 14000]
-
-  WARMUP_ITERS: 2000
-
 DATASETS:
   NAMES: ("Market1501",)
   TESTS: ("Market1501",)

configs/Market1501/bagtricks_R101-ibn.yml

@@ -9,12 +9,6 @@ MODEL:
   HEADS:
     NUM_CLASSES: 751
 
-SOLVER:
-  MAX_ITER: 18000
-
-  STEPS: [8000, 14000]
-
-  WARMUP_ITERS: 2000
 
 DATASETS:
   NAMES: ("Market1501",)

configs/Market1501/bagtricks_R50-ibn.yml

@@ -8,12 +8,6 @@ MODEL:
   HEADS:
     NUM_CLASSES: 751
 
-SOLVER:
-  MAX_ITER: 18000
-
-  STEPS: [8000, 14000]
-
-  WARMUP_ITERS: 2000
 
 DATASETS:
   NAMES: ("Market1501",)

configs/Market1501/bagtricks_R50.yml

@@ -4,15 +4,10 @@ MODEL:
   HEADS:
     NUM_CLASSES: 751
 
-SOLVER:
-  MAX_ITER: 18000
-
-  STEPS: [8000, 14000]
-
-  WARMUP_ITERS: 2000
 
 DATASETS:
   NAMES: ("Market1501",)
   TESTS: ("Market1501",)
 
+
 OUTPUT_DIR: "logs/market1501/bagtricks_R50"

configs/Market1501/bagtricks_S50.yml

@@ -7,12 +7,6 @@ MODEL:
   HEADS:
     NUM_CLASSES: 751
 
-SOLVER:
-  MAX_ITER: 18000
-
-  STEPS: [8000, 14000]
-
-  WARMUP_ITERS: 2000
 
 DATASETS:
   NAMES: ("Market1501",)

configs/Market1501/sbs_R101-ibn.yml

@@ -10,13 +10,6 @@ MODEL:
   HEADS:
     NUM_CLASSES: 751
 
-SOLVER:
-  MAX_ITER: 18000
-
-  STEPS: [8000, 14000]
-
-  WARMUP_ITERS: 2000
-
 DATASETS:
   NAMES: ("Market1501",)
   TESTS: ("Market1501",)
@@ -25,4 +18,4 @@ TEST:
   PRECISE_BN:
     DATASET: 'Market1501'
 
-OUTPUT_DIR: "logs/market1501/bagtricks_R101-ibn"
+OUTPUT_DIR: "logs/market1501/sbs-R101-ibn"

configs/Market1501/sbs_R50-ibn.yml

@@ -3,17 +3,11 @@ _BASE_: "../Base-Strongerbaseline.yml"
 MODEL:
   BACKBONE:
     WITH_IBN: True
-    PRETRAIN_PATH: "/export/home/lxy/.cache/torch/checkpoints/resnet50_ibn_a.pth.tar"
+    PRETRAIN_PATH: "/home/liaoxingyu2/lxy/.cache/torch/checkpoints/resnet50_ibn_a.pth.tar"
 
   HEADS:
     NUM_CLASSES: 751
 
-SOLVER:
-  MAX_ITER: 16000
-  DELAY_ITERS: 8000
-  WARMUP_ITERS: 2600
-  FREEZE_ITERS: 2600
-
 DATASETS:
   NAMES: ("Market1501",)
   TESTS: ("Market1501",)

configs/Market1501/sbs_R50.yml

@@ -4,12 +4,6 @@ MODEL:
   HEADS:
     NUM_CLASSES: 751
 
-SOLVER:
-  MAX_ITER: 16000
-  DELAY_ITERS: 8000
-  WARMUP_ITERS: 2600
-  FREEZE_ITERS: 2600
-
 DATASETS:
   NAMES: ("Market1501",)
   TESTS: ("Market1501",)

configs/Market1501/sbs_S50.yml

@@ -7,11 +7,9 @@ MODEL:
   HEADS:
     NUM_CLASSES: 751
 
-SOLVER:
+  LOSSES:
-  MAX_ITER: 16000
+    TRI:
-  DELAY_ITERS: 8000
+      MARGIN: 0.3
-  WARMUP_ITERS: 2600
-  FREEZE_ITERS: 2600
 
 DATASETS:
   NAMES: ("Market1501",)