hezar.models package

Subpackages

• hezar.models.backbone package
  • Subpackages
    • hezar.models.backbone.bert package
      • Submodules
      • Module contents
    • hezar.models.backbone.distilbert package
      • Submodules
      • Module contents
    • hezar.models.backbone.roberta package
      • Submodules
      • Module contents
    • hezar.models.backbone.vit package
      • Submodules
      • Module contents
  • Module contents
• hezar.models.image2text package
  • Subpackages
    • hezar.models.image2text.beit_roberta package
      • Submodules
      • Module contents
    • hezar.models.image2text.crnn package
      • Submodules
      • Module contents
    • hezar.models.image2text.trocr package
      • Submodules
      • Module contents
    • hezar.models.image2text.vit_gpt2 package
      • Submodules
      • Module contents
    • hezar.models.image2text.vit_roberta package
      • Submodules
      • Module contents
  • Module contents
• hezar.models.mask_filling package
  • Subpackages
    • hezar.models.mask_filling.bert package
      • Submodules
      • Module contents
    • hezar.models.mask_filling.distilbert package
      • Submodules
      • Module contents
    • hezar.models.mask_filling.roberta package
      • Submodules
      • Module contents
  • Module contents
• hezar.models.sequence_labeling package
  • Subpackages
    • hezar.models.sequence_labeling.bert package
      • Submodules
      • Module contents
    • hezar.models.sequence_labeling.distilbert package
      • Submodules
      • Module contents
    • hezar.models.sequence_labeling.roberta package
      • Submodules
      • Module contents
  • Module contents
• hezar.models.speech_recognition package
  • Subpackages
    • hezar.models.speech_recognition.whisper package
      • Submodules
      • Module contents
  • Module contents
• hezar.models.text_classification package
  • Subpackages
    • hezar.models.text_classification.bert package
      • Submodules
      • Module contents
    • hezar.models.text_classification.distilbert package
      • Submodules
      • Module contents
    • hezar.models.text_classification.roberta package
      • Submodules
      • Module contents
  • Module contents
• hezar.models.text_detection package
  • Subpackages
    • hezar.models.text_detection.craft package
      • Submodules
      • Module contents
  • Module contents
• hezar.models.text_generation package
  • Subpackages
    • hezar.models.text_generation.gpt2 package
      • Submodules
      • Module contents
    • hezar.models.text_generation.t5 package
      • Submodules
      • Module contents
  • Module contents

Submodules

• hezar.models.model module
  • Model
    • Model.compute_loss()
    • Model.config_filename
    • Model.device
    • Model.forward()
    • Model.generate()
    • Model.is_generative
    • Model.load()
    • Model.load_state_dict()
    • Model.loss_func
    • Model.loss_func_kwargs
    • Model.loss_func_name
    • Model.model_filename
    • Model.num_parameters
    • Model.num_trainable_parameters
    • Model.post_process()
    • Model.predict()
    • Model.preprocess()
    • Model.preprocessor
    • Model.push_to_hub()
    • Model.required_backends
    • Model.save()
    • Model.skip_keys_on_load
• hezar.models.model_outputs module
  • Image2TextOutput
    • Image2TextOutput.score
    • Image2TextOutput.text
  • MaskFillingOutput
    • MaskFillingOutput.score
    • MaskFillingOutput.sequence
    • MaskFillingOutput.token
    • MaskFillingOutput.token_id
  • ModelOutput
    • ModelOutput.dict()
    • ModelOutput.items()
    • ModelOutput.keys()
    • ModelOutput.values()
  • SequenceLabelingOutput
    • SequenceLabelingOutput.end
    • SequenceLabelingOutput.label
    • SequenceLabelingOutput.score
    • SequenceLabelingOutput.start
    • SequenceLabelingOutput.token
  • SpeechRecognitionOutput
    • SpeechRecognitionOutput.chunks
    • SpeechRecognitionOutput.text
  • TextClassificationOutput
    • TextClassificationOutput.label
    • TextClassificationOutput.score
  • TextDetectionOutput
    • TextDetectionOutput.boxes
  • TextGenerationOutput
    • TextGenerationOutput.text

Module contents