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
f7dcf8fc
"model_cards/vscode:/vscode.git/clone" did not exist on "ff82a2aa93e840547c8c91c6772ccc64af5b490c"
Commit
f7dcf8fc
authored
Mar 23, 2020
by
Julien Chaumond
Browse files
[BertAbs] Move files around for more consistent naming
parent
e25c4f40
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
3 additions
and
3 deletions
+3
-3
examples/summarization/bertabs/configuration_bertabs.py
examples/summarization/bertabs/configuration_bertabs.py
+1
-1
examples/summarization/bertabs/convert_bertabs_original_pytorch_checkpoint.py
...on/bertabs/convert_bertabs_original_pytorch_checkpoint.py
+1
-1
examples/summarization/bertabs/modeling_bertabs.py
examples/summarization/bertabs/modeling_bertabs.py
+1
-1
No files found.
examples/summarization/bertabs/configuration_bertabs.py
View file @
f7dcf8fc
...
...
@@ -23,7 +23,7 @@ logger = logging.getLogger(__name__)
BERTABS_FINETUNED_CONFIG_MAP
=
{
"bertabs-finetuned-cnndm"
:
"https://s3.amazonaws.com/models.huggingface.co/bert/remi/bertabs-finetuned-cnndm-extractive-abstractive-summarization
-
config.json"
,
"bertabs-finetuned-cnndm"
:
"https://s3.amazonaws.com/models.huggingface.co/bert/remi/bertabs-finetuned-cnndm-extractive-abstractive-summarization
/
config.json"
,
}
...
...
examples/summarization/bertabs/convert_bertabs_original_pytorch_checkpoint.py
View file @
f7dcf8fc
...
...
@@ -157,7 +157,7 @@ def convert_bertabs_checkpoints(path_to_checkpoints, dump_path):
# directory structure. We save the state_dict instead.
logging
.
info
(
"saving the model's state dictionary"
)
torch
.
save
(
new_model
.
state_dict
(),
"bertabs-finetuned-cnndm-extractive-abstractive-summarization
-
pytorch_model.bin"
new_model
.
state_dict
(),
"
./
bertabs-finetuned-cnndm-extractive-abstractive-summarization
/
pytorch_model.bin"
)
...
...
examples/summarization/bertabs/modeling_bertabs.py
View file @
f7dcf8fc
...
...
@@ -34,7 +34,7 @@ from transformers import BertConfig, BertModel, PreTrainedModel
MAX_SIZE
=
5000
BERTABS_FINETUNED_MODEL_MAP
=
{
"bertabs-finetuned-cnndm"
:
"https://s3.amazonaws.com/models.huggingface.co/bert/remi/bertabs-finetuned-cnndm-extractive-abstractive-summarization
-
pytorch_model.bin"
,
"bertabs-finetuned-cnndm"
:
"https://s3.amazonaws.com/models.huggingface.co/bert/remi/bertabs-finetuned-cnndm-extractive-abstractive-summarization
/
pytorch_model.bin"
,
}
...
...
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