flambe.nlp.transformers
¶
Submodules¶
Package Contents¶
-
class
flambe.nlp.transformers.
BertTextField
[source]¶ Bases:
flambe.nlp.transformers.utils.TransformerTextField
Integrate the pytorch_transformers BertTokenizer.
- Currently available aliases:
- . bert-base-uncased . bert-large-uncased . bert-base-cased . bert-large-cased . bert-base-multilingual-uncased . bert-base-multilingual-cased . bert-base-chinese . bert-base-german-cased . bert-large-uncased-whole-word-masking . bert-large-cased-whole-word-masking . bert-large-uncased-whole-word-masking-finetuned-squad . bert-large-cased-whole-word-masking-finetuned-squad . bert-base-cased-finetuned-mrpc
-
_cls
¶
-
class
flambe.nlp.transformers.
BertEmbedder
[source]¶ Bases:
flambe.nlp.transformers.utils.TransformerEmbedder
Integrate the pytorch_transformers BertModel.
- Currently available aliases:
- . bert-base-uncased . bert-large-uncased . bert-base-cased . bert-large-cased . bert-base-multilingual-uncased . bert-base-multilingual-cased . bert-base-chinese . bert-base-german-cased . bert-large-uncased-whole-word-masking . bert-large-cased-whole-word-masking . bert-large-uncased-whole-word-masking-finetuned-squad . bert-large-cased-whole-word-masking-finetuned-squad . bert-base-cased-finetuned-mrpc
-
_cls
¶
-
class
flambe.nlp.transformers.
RobertaTextField
[source]¶ Bases:
flambe.nlp.transformers.utils.TransformerTextField
Integrate the pytorch_transformers RobertaTokenizer.
- Currently available aliases:
- . roberta-base . roberta-large . roberta-large-mnli
-
_cls
¶
-
class
flambe.nlp.transformers.
RobertaEmbedder
[source]¶ Bases:
flambe.nlp.transformers.utils.TransformerEmbedder
Integrate the pytorch_transformers RobertaModel.
- Currently available aliases:
- . roberta-base . roberta-large . roberta-large-mnli
-
_cls
¶
-
class
flambe.nlp.transformers.
GPTTextField
[source]¶ Bases:
flambe.nlp.transformers.utils.TransformerTextField
Integrate the pytorch_transformers OpenAIGPTTokenizer.
- Currently available aliases:
- . openai-gpt
-
_cls
¶
-
class
flambe.nlp.transformers.
GPTEmbedder
[source]¶ Bases:
flambe.nlp.transformers.utils.TransformerEmbedder
Integrate the pytorch_transformers OpenAIGPTmodel.
- Currently available aliases:
- . openai-gpt
-
_cls
¶
-
class
flambe.nlp.transformers.
GPT2TextField
[source]¶ Bases:
flambe.nlp.transformers.utils.TransformerTextField
Integrate the pytorch_transformers GPT2Tokenizer.
- Currently available aliases:
- . gpt2 . gpt2-medium . gpt2-large
-
_cls
¶
-
class
flambe.nlp.transformers.
GPT2Embedder
[source]¶ Bases:
flambe.nlp.transformers.utils.TransformerEmbedder
Integrate the pytorch_transformers GPT2Model.
- Currently available aliases:
- . gpt2 . gpt2-medium . gpt2-large
-
_cls
¶
-
class
flambe.nlp.transformers.
XLMTextField
[source]¶ Bases:
flambe.nlp.transformers.utils.TransformerTextField
Integrate the pytorch_transformers XLMTokenizer.
- Currently available aliases:
- . xlm-mlm-en-2048 . xlm-mlm-ende-1024 . xlm-mlm-enfr-1024 . xlm-mlm-enro-1024 . xlm-mlm-tlm-xnli15-1024 . xlm-mlm-xnli15-1024 . xlm-clm-enfr-1024 . xlm-clm-ende-1024
-
_cls
¶
-
class
flambe.nlp.transformers.
XLMEmbedder
[source]¶ Bases:
flambe.nlp.transformers.utils.TransformerEmbedder
Integrate the pytorch_transformers XLMModel.
- Currently available aliases:
- . xlm-mlm-en-2048 . xlm-mlm-ende-1024 . xlm-mlm-enfr-1024 . xlm-mlm-enro-1024 . xlm-mlm-tlm-xnli15-1024 . xlm-mlm-xnli15-1024 . xlm-clm-enfr-1024 . xlm-clm-ende-1024
-
_cls
¶
-
class
flambe.nlp.transformers.
XLNetTextField
[source]¶ Bases:
flambe.nlp.transformers.utils.TransformerTextField
Integrate the pytorch_transformers XLNetTokenizer.
- Currently available aliases:
- . xlnet-base-cased . xlnet-large-cased
-
_cls
¶
-
class
flambe.nlp.transformers.
XLNetEmbedder
[source]¶ Bases:
flambe.nlp.transformers.utils.TransformerEmbedder
Integrate the pytorch_transformers XLNetModel.
- Currently available aliases:
- . xlnet-base-cased . xlnet-large-cased
-
_cls
¶
-
class
flambe.nlp.transformers.
TransfoXLTextField
[source]¶ Bases:
flambe.nlp.transformers.utils.TransformerTextField
Integrate the pytorch_transformers TransfoXLTokenizer.
- Currently available aliases:
- . transfo-xl-wt103
-
_cls
¶
-
class
flambe.nlp.transformers.
TransfoXLEmbedder
[source]¶ Bases:
flambe.nlp.transformers.utils.TransformerEmbedder
Integrate the pytorch_transformers TransfoXLModel.
- Currently available aliases:
- . transfo-xl-wt103
-
_cls
¶
-
class
flambe.nlp.transformers.
AdamW
[source]¶ Bases:
flambe.Component
,pytorch_transformers.AdamW
-
class
flambe.nlp.transformers.
ConstantLRSchedule
[source]¶ Bases:
flambe.Component
,pytorch_transformers.ConstantLRSchedule
-
class
flambe.nlp.transformers.
WarmupConstantSchedule
[source]¶ Bases:
flambe.Component
,pytorch_transformers.WarmupConstantSchedule
-
class
flambe.nlp.transformers.
WarmupLinearSchedule
[source]¶ Bases:
flambe.Component
,pytorch_transformers.WarmupLinearSchedule