diff --git a/fastreid/config/defaults.py b/fastreid/config/defaults.py index 2ae1cfc..505b5c6 100644 --- a/fastreid/config/defaults.py +++ b/fastreid/config/defaults.py @@ -163,9 +163,9 @@ _C.MODEL.PCB.HEAD.EMBEDDING_DIM = 512 # ----------------------------------------------------------------------------- _C.INPUT = CN() # Size of the image during training -_C.INPUT.SIZE_TRAIN = [256, 128] +_C.INPUT.SIZE_TRAIN = [0, 0] # Size of the image during test -_C.INPUT.SIZE_TEST = [256, 128] +_C.INPUT.SIZE_TEST = [0, 0] # `True` if cropping is used for data augmentation during training _C.INPUT.CROP = CN({"ENABLED": False}) diff --git a/projects/FastShoe/configs/base.yaml b/projects/FastShoe/configs/base.yaml new file mode 100644 index 0000000..f4cca7c --- /dev/null +++ b/projects/FastShoe/configs/base.yaml @@ -0,0 +1,63 @@ +INPUT: + SIZE_TRAIN: [0, 0] # resize size + SIZE_TEST: [0, 0] + + CROP: + ENABLED: False + + FLIP: + ENABLED: False + + PADDING: + ENABLED: False + + CJ: + ENABLED: False + + AFFINE: + ENABLED: False + + AUTOAUG: + ENABLED: False + + AUGMIX: + ENABLED: False + + REA: + ENABLED: False + + RPT: + ENABLED: False + + +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 diff --git a/projects/FastShoe/configs/r101-se-ibn.yaml b/projects/FastShoe/configs/r101-se-ibn.yaml deleted file mode 100644 index 2f3320b..0000000 --- a/projects/FastShoe/configs/r101-se-ibn.yaml +++ /dev/null @@ -1,89 +0,0 @@ -_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: 16 - -SOLVER: - MAX_EPOCH: 300 - AMP: - ENABLED: True - - OPT: SGD - SCHED: CosineAnnealingLR - - BASE_LR: 0.001 - MOMENTUM: 0.99 - NESTEROV: False - - BIAS_LR_FACTOR: 1. - WEIGHT_DECAY: 0.0005 - WEIGHT_DECAY_BIAS: 0. - IMS_PER_BATCH: 4 - - ETA_MIN_LR: 0.00003 - - WARMUP_FACTOR: 0.1 - WARMUP_ITERS: 100 - - CHECKPOINT_PERIOD: 1 - -TEST: - EVAL_PERIOD: 1 - IMS_PER_BATCH: 128 - -DATASETS: - NAMES: ("ShoeDataset",) - TESTS: ("ShoeDataset", "OnlineDataset") - -OUTPUT_DIR: projects/FastShoe/logs/r101_ibn_se