hezar.models.sequence_labeling.roberta package

Submodules

• hezar.models.sequence_labeling.roberta.roberta_sequence_labeling module
  • RobertaClassificationHead
    • RobertaClassificationHead.forward()
  • RobertaSequenceLabeling
    • RobertaSequenceLabeling.compute_loss()
    • RobertaSequenceLabeling.forward()
    • RobertaSequenceLabeling.post_process()
    • RobertaSequenceLabeling.preprocess()
    • RobertaSequenceLabeling.required_backends
    • RobertaSequenceLabeling.skip_keys_on_load
• hezar.models.sequence_labeling.roberta.roberta_sequence_labeling_config module
  • RobertaSequenceLabelingConfig
    • RobertaSequenceLabelingConfig.attention_probs_dropout_prob
    • RobertaSequenceLabelingConfig.bos_token_id
    • RobertaSequenceLabelingConfig.classifier_dropout
    • RobertaSequenceLabelingConfig.eos_token_id
    • RobertaSequenceLabelingConfig.gradient_checkpointing
    • RobertaSequenceLabelingConfig.hidden_act
    • RobertaSequenceLabelingConfig.hidden_dropout_prob
    • RobertaSequenceLabelingConfig.hidden_size
    • RobertaSequenceLabelingConfig.id2label
    • RobertaSequenceLabelingConfig.initializer_range
    • RobertaSequenceLabelingConfig.intermediate_size
    • RobertaSequenceLabelingConfig.layer_norm_eps
    • RobertaSequenceLabelingConfig.max_position_embeddings
    • RobertaSequenceLabelingConfig.name
    • RobertaSequenceLabelingConfig.num_attention_heads
    • RobertaSequenceLabelingConfig.num_hidden_layers
    • RobertaSequenceLabelingConfig.num_labels
    • RobertaSequenceLabelingConfig.pad_token_id
    • RobertaSequenceLabelingConfig.position_embedding_type
    • RobertaSequenceLabelingConfig.prediction_skip_tokens
    • RobertaSequenceLabelingConfig.task
    • RobertaSequenceLabelingConfig.type_vocab_size
    • RobertaSequenceLabelingConfig.use_cache
    • RobertaSequenceLabelingConfig.vocab_size

Module contents