hezar.models.mask_filling.roberta package

Submodules

• hezar.models.mask_filling.roberta.roberta_mask_filling module
  • RobertaMaskFilling
    • RobertaMaskFilling.compute_loss()
    • RobertaMaskFilling.forward()
    • RobertaMaskFilling.loss_func_name
    • RobertaMaskFilling.post_process()
    • RobertaMaskFilling.preprocess()
    • RobertaMaskFilling.required_backends
    • RobertaMaskFilling.skip_keys_on_load
    • RobertaMaskFilling.tokenizer_name
• hezar.models.mask_filling.roberta.roberta_mask_filling_config module
  • RobertaMaskFillingConfig
    • RobertaMaskFillingConfig.attention_probs_dropout_prob
    • RobertaMaskFillingConfig.bos_token_id
    • RobertaMaskFillingConfig.eos_token_id
    • RobertaMaskFillingConfig.gradient_checkpointing
    • RobertaMaskFillingConfig.hidden_act
    • RobertaMaskFillingConfig.hidden_dropout_prob
    • RobertaMaskFillingConfig.hidden_size
    • RobertaMaskFillingConfig.initializer_range
    • RobertaMaskFillingConfig.intermediate_size
    • RobertaMaskFillingConfig.layer_norm_eps
    • RobertaMaskFillingConfig.max_position_embeddings
    • RobertaMaskFillingConfig.name
    • RobertaMaskFillingConfig.num_attention_heads
    • RobertaMaskFillingConfig.num_hidden_layers
    • RobertaMaskFillingConfig.pad_token_id
    • RobertaMaskFillingConfig.position_embedding_type
    • RobertaMaskFillingConfig.task
    • RobertaMaskFillingConfig.type_vocab_size
    • RobertaMaskFillingConfig.use_cache
    • RobertaMaskFillingConfig.vocab_size

Module contents