diff --git a/projects/FastShoe/configs/base-pair.yaml b/projects/FastShoe/configs/base-pair.yaml index 2841b46..dd09fbf 100644 --- a/projects/FastShoe/configs/base-pair.yaml +++ b/projects/FastShoe/configs/base-pair.yaml @@ -38,7 +38,7 @@ INPUT: ENABLED: True CROP: - ENABLED: True + ENABLED: False SIZE: [224,] SCALE: [0.08, 1] RATIO: [0.75, 1.333333333] diff --git a/projects/FastShoe/configs/r101-se-ibn.yaml b/projects/FastShoe/configs/r101-se-ibn.yaml new file mode 100644 index 0000000..ca005bf --- /dev/null +++ b/projects/FastShoe/configs/r101-se-ibn.yaml @@ -0,0 +1,89 @@ +_BASE_: base-pair.yaml + +MODEL: + META_ARCHITECTURE: Metric + + BACKBONE: + NAME: build_resnet_backbone + DEPTH: 101x + NORM: BN + LAST_STRIDE: 2 + FEAT_DIM: 512 + PRETRAIN: True + WITH_IBN: True + WITH_SE: True + + HEADS: + NAME: PairHead + WITH_BNNECK: False + EMBEDDING_DIM: 0 + POOL_LAYER: FastGlobalAvgPool + CLS_LAYER: Linear + NUM_CLASSES: 2 + + LOSSES: + NAME: ("ContrastiveLoss",) + + CONTRASTIVE: + MARGIN: 2.0 + SCALE: 1. + +INPUT: + SIZE_TRAIN: [224, 224] # no need for resize when training + SIZE_TEST: [224, 224] + + FLIP: + ENABLED: True + + CJ: + ENABLED: True + + AUTOAUG: + ENABLED: True + PROB: 0.5 + + AUGMIX: + ENABLED: True + PROB: 0.5 + + REA: + ENABLED: True + + +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: 16 + + ETA_MIN_LR: 0.00003 + + WARMUP_FACTOR: 0.1 + WARMUP_ITERS: 100 + + CHECKPOINT_PERIOD: 1 + +TEST: + EVAL_PERIOD: 1 + IMS_PER_BATCH: 32 + +DATASETS: + NAMES: ("ShoeDataset",) + TESTS: ("ShoeDataset", "OnlineDataset") + +OUTPUT_DIR: projects/FastShoe/logs/r101_ibn_se