Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "049e14f0e3079a686223297bb7d5c7862eafda0d"
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 Hide 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 ...@@ -10,7 +10,8 @@ from .tokenization_roberta import RobertaTokenizer
from .tokenization_utils import (PreTrainedTokenizer) 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, from .modeling_bert import (BertConfig, BertPreTrainedModel, BertModel, BertForPreTraining,
BertForMaskedLM, BertForNextSentencePrediction, 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