Base-AGW.yml 285 Bytes
Newer Older
dengjb's avatar
update  
dengjb committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
_BASE_: Base-bagtricks.yml

MODEL:
  BACKBONE:
    WITH_NL: True

  HEADS:
    POOL_LAYER: GeneralizedMeanPooling

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

    TRI:
      MARGIN: 0.0
      HARD_MINING: False
      SCALE: 1.0