Skip to content

GitLab

Unverified Commit 9bdce3a4 authored by Sam Shleifer's avatar Sam Shleifer Committed by GitHub
Browse files

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

parent de4d7b00
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/seq2seq/make_student.py
View file @ 9bdce3a4
...@@ -32,7 +32,7 @@ LAYERS_TO_COPY = { ...@@ -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],
......
examples/seq2seq/sentence_splitter.py
View file @ 9bdce3a4
...@@ -11,7 +11,7 @@ except (ImportError, ModuleNotFoundError): ...@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment