Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "8c67b529f615cc24c46864b8323d2d47a15ccd58"
Commit cb60ce59 authored by LysandreJik's avatar LysandreJik
Browse files

Added multiple AutoModel classes: AutoModelWithLMHead,... 

Added multiple AutoModel classes: AutoModelWithLMHead, AutoModelForQuestionAnswering and AutoModelForSequenceClassification
parent df9d6eff
Expand all Show whitespace changes
Inline Side-by-side
Showing with 344 additions and 18 deletions
pytorch_transformers/__init__.py
View file @ cb60ce59
......@@ -10,7 +10,8 @@ from .tokenization_roberta import RobertaTokenizer
from .tokenization_utils import (PreTrainedTokenizer)
from .modeling_auto import (AutoConfig, AutoModel)
from .modeling_auto import (AutoConfig, AutoModel, AutoModelForSequenceClassification, AutoModelForQuestionAnswering,
AutoModelWithLMHead)
from .modeling_bert import (BertConfig, BertPreTrainedModel, BertModel, BertForPreTraining,
BertForMaskedLM, BertForNextSentencePrediction,
......
pytorch_transformers/modeling_auto.py
View file @ cb60ce59
This diff is collapsed.
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