From 0fa19dca61d413a802c93b3fc284e67d3bcb6634 Mon Sep 17 00:00:00 2001
From: Felix <freehuang100@163.com>
Date: Thu, 17 Jun 2021 16:33:34 +0800
Subject: [PATCH] Update ResNet50_ReID.yaml

---
 ppcls/configs/Logo/ResNet50_ReID.yaml | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/ppcls/configs/Logo/ResNet50_ReID.yaml b/ppcls/configs/Logo/ResNet50_ReID.yaml
index 19ad20b5..99598ba1 100644
--- a/ppcls/configs/Logo/ResNet50_ReID.yaml
+++ b/ppcls/configs/Logo/ResNet50_ReID.yaml
@@ -19,6 +19,8 @@ Global:
 # model architecture
 Arch:
   name: "RecModel"
+  infer_output_key: "features"
+  infer_add_softmax: False
   Backbone:
     name: "ResNet50_last_stage_stride1"
     pretrained: True
-- 
GitLab