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

davewang · Oct 2, 2020 · 9bdce3a · 9bdce3a
1 parent de4d7b0
commit 9bdce3a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/seq2seq/make_student.py b/examples/seq2seq/make_student.py
@@ -32,7 +32,7 @@ def copy_layers(src_layers: nn.ModuleList, dest_layers: nn.ModuleList, layers_to
     },
     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
@@ -11,7 +11,7 @@
     NLTK_AVAILABLE = False
 
 if NLTK_AVAILABLE:
-    with FileLock("a_random_string") as lock:
+    with FileLock(".lock") as lock:
         nltk.download("punkt", quiet=True)