hezar.models.mask_filling.bert package

Submodules

• hezar.models.mask_filling.bert.bert_mask_filling module
  • BertMaskFilling
    • BertMaskFilling.compute_loss()
    • BertMaskFilling.forward()
    • BertMaskFilling.loss_func_name
    • BertMaskFilling.post_process()
    • BertMaskFilling.preprocess()
    • BertMaskFilling.required_backends
    • BertMaskFilling.skip_keys_on_load
• hezar.models.mask_filling.bert.bert_mask_filling_config module
  • BertMaskFillingConfig
    • BertMaskFillingConfig.attention_probs_dropout_prob
    • BertMaskFillingConfig.classifier_dropout
    • BertMaskFillingConfig.hidden_act
    • BertMaskFillingConfig.hidden_dropout_prob
    • BertMaskFillingConfig.hidden_size
    • BertMaskFillingConfig.initializer_range
    • BertMaskFillingConfig.intermediate_size
    • BertMaskFillingConfig.layer_norm_eps
    • BertMaskFillingConfig.max_position_embeddings
    • BertMaskFillingConfig.name
    • BertMaskFillingConfig.num_attention_heads
    • BertMaskFillingConfig.num_hidden_layers
    • BertMaskFillingConfig.pad_token_id
    • BertMaskFillingConfig.position_embedding_type
    • BertMaskFillingConfig.task
    • BertMaskFillingConfig.type_vocab_size
    • BertMaskFillingConfig.use_cache
    • BertMaskFillingConfig.vocab_size

Module contents