easydel.modules.roberta.modeling_roberta_flax

Contents

easydel.modules.roberta.modeling_roberta_flax#

class easydel.modules.roberta.modeling_roberta_flax.RobertaAttention(*args: Any, **kwargs: Any)[source]#: Bases: Module

class easydel.modules.roberta.modeling_roberta_flax.RobertaClassificationHead(*args: Any, **kwargs: Any)[source]#: Bases: Module

class easydel.modules.roberta.modeling_roberta_flax.RobertaEmbeddings(*args: Any, **kwargs: Any)[source]#: Bases: Module

class easydel.modules.roberta.modeling_roberta_flax.RobertaEncoder(*args: Any, **kwargs: Any)[source]#: Bases: Module

class easydel.modules.roberta.modeling_roberta_flax.RobertaForCausalLM(*args: Any, **kwargs: Any)[source]#: Bases: EasyDeLBaseModule

class easydel.modules.roberta.modeling_roberta_flax.RobertaForMultipleChoice(*args: Any, **kwargs: Any)[source]#: Bases: EasyDeLBaseModule

class easydel.modules.roberta.modeling_roberta_flax.RobertaForQuestionAnswering(*args: Any, **kwargs: Any)[source]#: Bases: EasyDeLBaseModule

class easydel.modules.roberta.modeling_roberta_flax.RobertaForSequenceClassification(*args: Any, **kwargs: Any)[source]#: Bases: EasyDeLBaseModule

class easydel.modules.roberta.modeling_roberta_flax.RobertaForTokenClassification(*args: Any, **kwargs: Any)[source]#: Bases: EasyDeLBaseModule

class easydel.modules.roberta.modeling_roberta_flax.RobertaIntermediate(*args: Any, **kwargs: Any)[source]#: Bases: Module

class easydel.modules.roberta.modeling_roberta_flax.RobertaLMHead(*args: Any, **kwargs: Any)[source]#: Bases: Module

class easydel.modules.roberta.modeling_roberta_flax.RobertaLayer(*args: Any, **kwargs: Any)[source]#: Bases: Module

class easydel.modules.roberta.modeling_roberta_flax.RobertaModel(*args: Any, **kwargs: Any)[source]#: Bases: EasyDeLBaseModule

class easydel.modules.roberta.modeling_roberta_flax.RobertaOutput(*args: Any, **kwargs: Any)[source]#: Bases: Module

class easydel.modules.roberta.modeling_roberta_flax.RobertaPooler(*args: Any, **kwargs: Any)[source]#: Bases: Module

class easydel.modules.roberta.modeling_roberta_flax.RobertaSelfAttention(*args: Any, **kwargs: Any)[source]#: Bases: AttentionModule

class easydel.modules.roberta.modeling_roberta_flax.RobertaSelfOutput(*args: Any, **kwargs: Any)[source]#: Bases: Module