hezar.models.mask_filling.distilbert package

Submodules

• hezar.models.mask_filling.distilbert.distilbert_mask_filling module
  • DistilBertMaskFilling
    • DistilBertMaskFilling.compute_loss()
    • DistilBertMaskFilling.forward()
    • DistilBertMaskFilling.loss_func_name
    • DistilBertMaskFilling.post_process()
    • DistilBertMaskFilling.preprocess()
    • DistilBertMaskFilling.required_backends
    • DistilBertMaskFilling.tokenizer_name
• hezar.models.mask_filling.distilbert.distilbert_mask_filling_config module
  • DistilBertMaskFillingConfig
    • DistilBertMaskFillingConfig.activation
    • DistilBertMaskFillingConfig.attention_dropout
    • DistilBertMaskFillingConfig.dim
    • DistilBertMaskFillingConfig.dropout
    • DistilBertMaskFillingConfig.hidden_dim
    • DistilBertMaskFillingConfig.initializer_range
    • DistilBertMaskFillingConfig.max_position_embeddings
    • DistilBertMaskFillingConfig.n_heads
    • DistilBertMaskFillingConfig.n_layers
    • DistilBertMaskFillingConfig.name
    • DistilBertMaskFillingConfig.output_past
    • DistilBertMaskFillingConfig.pad_token_id
    • DistilBertMaskFillingConfig.qa_dropout
    • DistilBertMaskFillingConfig.task
    • DistilBertMaskFillingConfig.tie_weights_
    • DistilBertMaskFillingConfig.vocab_size

Module contents