This commit is contained in:
littletomatodonkey
2022-05-06 13:38:42 +08:00
committed by GitHub
parent 896d149ed7
commit 8fd9347e18
@@ -129,7 +129,7 @@ Loss:
key: head_out
multi_head: True
- DistillationSARLoss:
weight: 0.5
weight: 1.0
model_name_list: ["Student", "Teacher"]
key: head_out
multi_head: True