diff --git a/projects/FastShoe/configs/online-pcb.yaml b/projects/FastShoe/configs/online-pcb.yaml index a6191a3..0afd488 100644 --- a/projects/FastShoe/configs/online-pcb.yaml +++ b/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 +