提交 662b0166 编写于 作者: littletomatodonkey's avatar littletomatodonkey

fix

上级 1dd898ae
...@@ -70,16 +70,14 @@ Loss: ...@@ -70,16 +70,14 @@ Loss:
mode: "l2" mode: "l2"
model_name_pairs: model_name_pairs:
- ["Student", "Teacher"] - ["Student", "Teacher"]
key: hidden_states key: hidden_states_5
index: 5
name: "loss_5" name: "loss_5"
- DistillationVQADistanceLoss: - DistillationVQADistanceLoss:
weight: 0.5 weight: 0.5
mode: "l2" mode: "l2"
model_name_pairs: model_name_pairs:
- ["Student", "Teacher"] - ["Student", "Teacher"]
key: hidden_states key: hidden_states_8
index: 8
name: "loss_8" name: "loss_8"
...@@ -182,4 +180,3 @@ Eval: ...@@ -182,4 +180,3 @@ Eval:
drop_last: False drop_last: False
batch_size_per_card: 8 batch_size_per_card: 8
num_workers: 4 num_workers: 4
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册