fast-reid/configs/Base-AGW.yml

_BASE_: "Base-bagtricks.yml"

MODEL:
  BACKBONE:
    WITH_NL: True

  HEADS:
    POOL_LAYER: "gempool"

  LOSSES:
    NAME: ("CrossEntropyLoss", "TripletLoss")
    CE:
      EPSILON: 0.1
      SCALE: 1.0

    TRI:
      MARGIN: 0.0
      HARD_MINING: False
      SCALE: 1.0