fix FC initialization in reid-strong-baseline baseline config

0c76d5a5 · HydrogenSulfate · 56b8710d · 0c76d5a5
隐藏空白更改
内联并排

Showing with 8 addition and 0 deletion

ppcls/configs/Pedestrian/strong_baseline_baseline.yaml ppcls/configs/Pedestrian/strong_baseline_baseline.yaml +8 -0

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