update online-pcb.yaml

projects/FastShoe/configs/online-pcb.yaml

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