update ser_vi_layoutxlm_xfund_zh_udml.yml

c8ca03c0 · 文幕地方 · 344c6d2d · c8ca03c0
隐藏空白更改
内联并排

Showing with 4 addition and 2 deletion

configs/kie/vi_layoutxlm/ser_vi_layoutxlm_xfund_zh_udml.yml configs/kie/vi_layoutxlm/ser_vi_layoutxlm_xfund_zh_udml.yml +4 -2

未找到文件。
--- a/configs/kie/vi_layoutxlm/ser_vi_layoutxlm_xfund_zh_udml.yml
+++ b/configs/kie/vi_layoutxlm/ser_vi_layoutxlm_xfund_zh_udml.yml
@@ -70,14 +70,16 @@ Loss:
      mode: "l2"
      model_name_pairs:
        - ["Student", "Teacher"]
-      key: hidden_states_5
+      key: hidden_states
+      index: 5
      name: "loss_5"
  - DistillationVQADistanceLoss:
      weight: 0.5
      mode: "l2"
      model_name_pairs:
        - ["Student", "Teacher"]
-      key: hidden_states_8
+      key: hidden_states
+      index: 8
      name: "loss_8"