From 0c76d5a52157fc9830722019cb21bab6013fb973 Mon Sep 17 00:00:00 2001
From: HydrogenSulfate <490868991@qq.com>
Date: Mon, 9 May 2022 15:14:01 +0800
Subject: [PATCH] fix FC initialization in reid-strong-baseline baseline config

---
 ppcls/configs/Pedestrian/strong_baseline_baseline.yaml | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/ppcls/configs/Pedestrian/strong_baseline_baseline.yaml b/ppcls/configs/Pedestrian/strong_baseline_baseline.yaml
index a0395f3b..ab4ba7b2 100644
--- a/ppcls/configs/Pedestrian/strong_baseline_baseline.yaml
+++ b/ppcls/configs/Pedestrian/strong_baseline_baseline.yaml
@@ -31,6 +31,14 @@ Arch:
     name: "FC"
     embedding_size: 2048
     class_num: 751
+    weight_attr:
+      initializer:
+        name: KaimingUniform
+        fan_in: 12288 # 6*embedding_size
+    bias_attr:
+      initializer:
+        name: KaimingUniform
+        fan_in: 12288 # 6*embedding_size
 
 # loss function config for traing/eval process
 Loss:
-- 
GitLab