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
5c2b94c8
"docs/vscode:/vscode.git/clone" did not exist on "da37a21c89c0555b6002b1fd51ee094e35a2e942"
Commit
5c2b94c8
authored
Aug 21, 2019
by
LysandreJik
Browse files
Changed string so that Circle CI accepts the warning
parent
87747518
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
2 deletions
+3
-2
pytorch_transformers/modeling_utils.py
pytorch_transformers/modeling_utils.py
+3
-2
No files found.
pytorch_transformers/modeling_utils.py
View file @
5c2b94c8
...
...
@@ -381,8 +381,9 @@ class PreTrainedModel(nn.Module):
self
.
config
.
pruned_heads
[
int
(
layer
)].
append
(
head
)
to_be_pruned
[
int
(
layer
)].
append
(
head
)
else
:
logger
.
warning
(
f
"Tried to remove head
{
head
}
of layer
{
layer
}
but it was already removed. "
f
"The removed heads are
{
heads_to_prune
}
"
)
logger
.
warning
(
"Tried to remove head "
+
head
+
" of layer "
+
layer
+
" but it was already removed. The current removed heads are "
+
heads_to_prune
)
base_model
.
_prune_heads
(
to_be_pruned
)
...
...
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