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

[s2s] fix lockfile and peg distillation constants (#7545)

Browse Source
This commit is contained in:
Sam Shleifer
2020-10-02 15:58:14 -04:00
committed by GitHub
parent de4d7b004a
commit 9bdce3a4f9
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/seq2seq/make_student.py
View File

@@ -32,7 +32,7 @@ LAYERS_TO_COPY = {
}, },
16: { # maps num layers in student -> which teacher layers to copy 16: { # maps num layers in student -> which teacher layers to copy
1: [0], 1: [0],
2: [0, 8], 2: [0, 15],
3: [0, 8, 15], 3: [0, 8, 15],
4: [0, 5, 10, 15], 4: [0, 5, 10, 15],
6: [0, 3, 6, 9, 12, 15], 6: [0, 3, 6, 9, 12, 15],

2
examples/seq2seq/sentence_splitter.py
View File

@@ -11,7 +11,7 @@ except (ImportError, ModuleNotFoundError):
NLTK_AVAILABLE = False NLTK_AVAILABLE = False
if NLTK_AVAILABLE: if NLTK_AVAILABLE:
with FileLock("a_random_string") as lock: with FileLock(".lock") as lock:
nltk.download("punkt", quiet=True) nltk.download("punkt", quiet=True)