hezar.models.backbone.distilbert package

Submodules

• hezar.models.backbone.distilbert.distilbert module
  • DistilBERT
    • DistilBERT.forward()
    • DistilBERT.post_process()
    • DistilBERT.preprocess()
    • DistilBERT.required_backends
• hezar.models.backbone.distilbert.distilbert_config module
  • DistilBERTConfig
    • DistilBERTConfig.activation
    • DistilBERTConfig.attention_dropout
    • DistilBERTConfig.dim
    • DistilBERTConfig.dropout
    • DistilBERTConfig.hidden_dim
    • DistilBERTConfig.initializer_range
    • DistilBERTConfig.max_position_embeddings
    • DistilBERTConfig.n_heads
    • DistilBERTConfig.n_layers
    • DistilBERTConfig.name
    • DistilBERTConfig.output_past
    • DistilBERTConfig.pad_token_id
    • DistilBERTConfig.qa_dropout
    • DistilBERTConfig.task
    • DistilBERTConfig.tie_weights_
    • DistilBERTConfig.vocab_size

Module contents