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

SOLVER:
  IMS_PER_BATCH: 64

MODEL_TEACHER:
  META_ARCHITECTURE: "Baseline"

  BACKBONE:
    NAME: "build_resnet_backbone"
    NORM: "BN"
    DEPTH: "50x"
    LAST_STRIDE: 1
    WITH_NL: True
    WITH_IBN: True
    PRETRAIN: False

  HEADS:
    NAME: "BNneckHead"
    IN_FEAT: 2048
    NORM: "BN"
    NECK_FEAT: "after"
    POOL_LAYER: "gempool"
    CLS_LAYER: "circle"
    SCALE: 64
    MARGIN: 0.35


MODEL:
  META_ARCHITECTURE: "Baseline"

  BACKBONE:
    WITH_NL: False
    DEPTH: "34x"
    WITH_IBN: True
  HEADS:
    IN_FEAT: 512
    CLS_LAYER: "circleSoftmax"

  STUDENT_WEIGHTS: ""
  TEACHER_WEIGHTS: "./logs/market1501/sbs_R50-ibn/model_final.pth"


DATASETS:
  NAMES: ("Market1501",)
  TESTS: ("Market1501",)

DATALOADER:
  NUM_WORKERS: 8

OUTPUT_DIR: "logs/market1501/sbs50ibn-kd-sbs34ibn"