SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[s2s] s/alpha_loss_encoder/alpha_encoder_loss/ (#7298)

Browse Source 
fix to match `distillation.py:        self.alpha_encoder_loss`
This commit is contained in:
Stas Bekman
2020-09-21 11:14:26 -07:00
committed by GitHub
parent cbb2f75a16
commit 8d562a2d1a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/seq2seq/test_seq2seq_examples.py
View File

@@ -86,7 +86,7 @@ CHEAP_ARGS = {
"n_val": -1, "n_val": -1,
"n_test": -1, "n_test": -1,
"student_encoder_layers": 1, "student_encoder_layers": 1,
"alpha_loss_encoder": 0.0, "alpha_encoder_loss": 0.0,
"freeze_encoder": False, "freeze_encoder": False,
"auto_scale_batch_size": False, "auto_scale_batch_size": False,
} }