From a5089348ee21a841cc2d76e5abea7ea3bdc0c509 Mon Sep 17 00:00:00 2001 From: weishengyu Date: Fri, 1 Apr 2022 11:41:02 +0800 Subject: [PATCH] dbg --- ppcls/configs/Pedestrian/reid_sb_ResNet50.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ppcls/configs/Pedestrian/reid_sb_ResNet50.yaml b/ppcls/configs/Pedestrian/reid_sb_ResNet50.yaml index 01f6cac8..af5e4ceb 100644 --- a/ppcls/configs/Pedestrian/reid_sb_ResNet50.yaml +++ b/ppcls/configs/Pedestrian/reid_sb_ResNet50.yaml @@ -24,7 +24,7 @@ Arch: name: "ResNet50" pretrained: True BackboneStopLayer: - name: "avg_pool" + name: "flatten" Head: name: "FC" embedding_size: 2048 @@ -36,7 +36,7 @@ Loss: - CELoss: weight: 1.0 epsilon: 0.1 - - TripletLoss: + - TripletLossV2: weight: 1.0 margin: 0.3 normalize_feature: false -- GitLab