Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
9bdce3a4
Unverified
Commit
9bdce3a4
authored
Oct 02, 2020
by
Sam Shleifer
Committed by
GitHub
Oct 02, 2020
Browse files
[s2s] fix lockfile and peg distillation constants (#7545)
parent
de4d7b00
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
2 deletions
+2
-2
examples/seq2seq/make_student.py
examples/seq2seq/make_student.py
+1
-1
examples/seq2seq/sentence_splitter.py
examples/seq2seq/sentence_splitter.py
+1
-1
No files found.
examples/seq2seq/make_student.py
View file @
9bdce3a4
...
...
@@ -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
],
...
...
examples/seq2seq/sentence_splitter.py
View file @
9bdce3a4
...
...
@@ -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
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment