From 78f2166596da6b7fbcdc9f7f659b72350ba954e7 Mon Sep 17 00:00:00 2001 From: liaoxingyu Date: Wed, 2 Sep 2020 18:01:07 +0800 Subject: [PATCH] refactor mgn config --- configs/Base-MGN.yml | 1 + configs/Market1501/mgn_R50-ibn.yml | 1 - 2 files changed, 1 insertion(+), 1 deletion(-) 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"