fast-reid/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot34ibn.yml

_BASE_: "../../../configs/Base-bagtricks.yml"

SOLVER:
  IMS_PER_BATCH: 64

MODEL_TEACHER:
  META_ARCHITECTURE: "Baseline"

  BACKBONE:
    NAME: "build_resnet_backbone"
    NORM: "BN"
    DEPTH: "101x"
    LAST_STRIDE: 1
    WITH_IBN: True
    PRETRAIN: True

  HEADS:
    NAME: "BNneckHead"
    IN_FEAT: 2048
    NORM: "BN"
    POOL_LAYER: "avgpool"
    NECK_FEAT: "before"
    CLS_LAYER: "linear"

MODEL:
  META_ARCHITECTURE: "Baseline"

  BACKBONE:
    NAME: "build_resnet_backbone"
    NORM: "BN"
    DEPTH: "34x"
    LAST_STRIDE: 1
    WITH_IBN: True
    PRETRAIN: True

  HEADS:
    NAME: "BNneckHead"
    IN_FEAT: 512
    NORM: "BN"
    POOL_LAYER: "avgpool"
    NECK_FEAT: "before"
    CLS_LAYER: "linear"

  STUDENT_WEIGHTS: ""
  TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth"
DATASETS:
  NAMES: ("Market1501",)
  TESTS: ("Market1501",)

DATALOADER:
  NUM_WORKERS: 8

OUTPUT_DIR: "logs/market1501/bot101ibn-kd-bot34ibn"