dbg

a5089348 · weishengyu · fb2ddcfd · a5089348
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

ppcls/configs/Pedestrian/reid_sb_ResNet50.yaml ppcls/configs/Pedestrian/reid_sb_ResNet50.yaml +2 -2

未找到文件。
--- 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