From 662b016694e37b929b3565a5ee6cf957cc796791 Mon Sep 17 00:00:00 2001 From: littletomatodonkey Date: Fri, 14 Oct 2022 13:39:28 +0000 Subject: [PATCH] fix --- .../kie/vi_layoutxlm/ser_vi_layoutxlm_xfund_zh_udml.yml | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/configs/kie/vi_layoutxlm/ser_vi_layoutxlm_xfund_zh_udml.yml b/configs/kie/vi_layoutxlm/ser_vi_layoutxlm_xfund_zh_udml.yml index 79abe540..51665337 100644 --- 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 - -- GitLab