fix

2022-10-14 13:39:28 +00:00 · 2022-10-14 13:39:28 +00:00 · 662b016694
parent 1dd898ae01
commit 662b016694
1 changed files with 2 additions and 5 deletions
--- 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,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
-