diff --git a/configs/Base-MGN.yml b/configs/Base-MGN.yml index f1bf5e4..be56202 100644 --- a/configs/Base-MGN.yml +++ b/configs/Base-MGN.yml @@ -7,6 +7,7 @@ MODEL: BACKBONE: WITH_NL: False + HEADS: NAME: "ReductionHead" POOL_LAYER: "maxpool" diff --git a/configs/Market1501/mgn_R50-ibn.yml b/configs/Market1501/mgn_R50-ibn.yml index b2ca6f2..2444c44 100644 --- a/configs/Market1501/mgn_R50-ibn.yml +++ b/configs/Market1501/mgn_R50-ibn.yml @@ -8,5 +8,4 @@ DATASETS: NAMES: ("Market1501",) TESTS: ("Market1501",) - OUTPUT_DIR: "logs/market1501/mgn_R50-ibn"