opensource/PaddleOCR
3
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2026-09-24 23:33:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #7936 from littletomatodonkey/dyg/fix_nlp_hidden_state

Browse Source 
fix hidden state for ser
This commit is contained in:
zhoujun
2022-10-14 22:14:01 +08:00
committed by GitHub
parent 1dd898ae01 662b016694
commit 8babfc86af
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/kie/vi_layoutxlm/ser_vi_layoutxlm_xfund_zh_udml.yml
+2 -5
View File
@@ -70,16 +70,14 @@ Loss:
mode: "l2"
model_name_pairs:
- ["Student", "Teacher"]
key: hidden_states
index: 5
key: hidden_states_5
name: "loss_5"
- DistillationVQADistanceLoss:
weight: 0.5
mode: "l2"
model_name_pairs:
- ["Student", "Teacher"]
key: hidden_states
index: 8
key: hidden_states_8
name: "loss_8"
@@ -182,4 +180,3 @@ Eval:
drop_last: False
batch_size_per_card: 8
num_workers: 4