From 9bdce3a4f91c6d53873582b0210e61c92bba8fd3 Mon Sep 17 00:00:00 2001 From: Sam Shleifer Date: Fri, 2 Oct 2020 15:58:14 -0400 Subject: [PATCH] [s2s] fix lockfile and peg distillation constants (#7545) --- examples/seq2seq/make_student.py | 2 +- examples/seq2seq/sentence_splitter.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/seq2seq/make_student.py b/examples/seq2seq/make_student.py index 6a19ef9aa1..32328363fe 100644 --- a/examples/seq2seq/make_student.py +++ b/examples/seq2seq/make_student.py @@ -32,7 +32,7 @@ LAYERS_TO_COPY = { }, 16: { # maps num layers in student -> which teacher layers to copy 1: [0], - 2: [0, 8], + 2: [0, 15], 3: [0, 8, 15], 4: [0, 5, 10, 15], 6: [0, 3, 6, 9, 12, 15], diff --git a/examples/seq2seq/sentence_splitter.py b/examples/seq2seq/sentence_splitter.py index 62fec13c3c..c5acec7392 100644 --- a/examples/seq2seq/sentence_splitter.py +++ b/examples/seq2seq/sentence_splitter.py @@ -11,7 +11,7 @@ except (ImportError, ModuleNotFoundError): NLTK_AVAILABLE = False if NLTK_AVAILABLE: - with FileLock("a_random_string") as lock: + with FileLock(".lock") as lock: nltk.download("punkt", quiet=True)