Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "146c521235ca057570cac4c1fc3f884ac464e580"
Commit 805c21ae authored by erenup's avatar erenup
Browse files

tried to fix the failed checks

parent d000195e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 4 deletions
transformers/modeling_roberta.py
View file @ 805c21ae
...@@ -591,10 +591,7 @@ class RobertaForQuestionAnswering(BertPreTrainedModel): ...@@ -591,10 +591,7 @@ class RobertaForQuestionAnswering(BertPreTrainedModel):
input_ids = tokenizer.encode(question, text) input_ids = tokenizer.encode(question, text)
start_scores, end_scores = model(torch.tensor([input_ids])) start_scores, end_scores = model(torch.tensor([input_ids]))
all_tokens = tokenizer.convert_ids_to_tokens(input_ids) all_tokens = tokenizer.convert_ids_to_tokens(input_ids)
print(' '.join(all_tokens[torch.argmax(start_scores) : torch.argmax(end_scores)+1])) answer = ' '.join(all_tokens[torch.argmax(start_scores) : torch.argmax(end_scores)+1])
a nice puppet
Note: 'roberta-large' model can not produce the right answer above. Waiting for 'roberta-large-finetuned-squad'
to be uploaded.
""" """
config_class = RobertaConfig config_class = RobertaConfig
pretrained_model_archive_map = ROBERTA_PRETRAINED_MODEL_ARCHIVE_MAP pretrained_model_archive_map = ROBERTA_PRETRAINED_MODEL_ARCHIVE_MAP
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment