diff --git a/ppcls/configs/Pedestrian/strong_baseline_baseline.yaml b/ppcls/configs/Pedestrian/strong_baseline_baseline.yaml index a0395f3b129bd0f2148e0e9cfd62dadaf8692ff9..ab4ba7b2021d6ed85152d79ebad747e02c67c377 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: