hezar.models.image2text.beit_roberta package

Submodules

• hezar.models.image2text.beit_roberta.beit_roberta_image2text module
  • BeitRobertaImage2Text
    • BeitRobertaImage2Text.compute_loss()
    • BeitRobertaImage2Text.forward()
    • BeitRobertaImage2Text.generate()
    • BeitRobertaImage2Text.image_processor
    • BeitRobertaImage2Text.is_generative
    • BeitRobertaImage2Text.loss_func_name
    • BeitRobertaImage2Text.post_process()
    • BeitRobertaImage2Text.preprocess()
    • BeitRobertaImage2Text.required_backends
    • BeitRobertaImage2Text.tokenizer_name
• hezar.models.image2text.beit_roberta.beit_roberta_image2text_config module
  • BeitRobertaImage2TextConfig
    • BeitRobertaImage2TextConfig.decoder
    • BeitRobertaImage2TextConfig.encoder
    • BeitRobertaImage2TextConfig.generation
    • BeitRobertaImage2TextConfig.name
  • DecoderConfig
    • DecoderConfig.add_cross_attention
    • DecoderConfig.attention_probs_dropout_prob
    • DecoderConfig.bos_token_id
    • DecoderConfig.classifier_dropout
    • DecoderConfig.eos_token_id
    • DecoderConfig.gradient_checkpointing
    • DecoderConfig.hidden_act
    • DecoderConfig.hidden_dropout_prob
    • DecoderConfig.hidden_size
    • DecoderConfig.initializer_range
    • DecoderConfig.intermediate_size
    • DecoderConfig.is_decoder
    • DecoderConfig.layer_norm_eps
    • DecoderConfig.max_position_embeddings
    • DecoderConfig.name
    • DecoderConfig.num_attention_heads
    • DecoderConfig.num_hidden_layers
    • DecoderConfig.pad_token_id
    • DecoderConfig.position_embedding_type
    • DecoderConfig.type_vocab_size
    • DecoderConfig.use_cache
    • DecoderConfig.vocab_size
  • EncoderConfig
    • EncoderConfig.attention_probs_dropout_prob
    • EncoderConfig.auxiliary_channels
    • EncoderConfig.auxiliary_concat_input
    • EncoderConfig.auxiliary_loss_weight
    • EncoderConfig.auxiliary_num_convs
    • EncoderConfig.drop_path_rate
    • EncoderConfig.hidden_act
    • EncoderConfig.hidden_dropout_prob
    • EncoderConfig.hidden_size
    • EncoderConfig.image_size
    • EncoderConfig.initializer_range
    • EncoderConfig.intermediate_size
    • EncoderConfig.layer_norm_eps
    • EncoderConfig.layer_scale_init_value
    • EncoderConfig.name
    • EncoderConfig.num_attention_heads
    • EncoderConfig.num_channels
    • EncoderConfig.num_hidden_layers
    • EncoderConfig.out_indices
    • EncoderConfig.patch_size
    • EncoderConfig.pool_scales
    • EncoderConfig.semantic_loss_ignore_index
    • EncoderConfig.use_absolute_position_embeddings
    • EncoderConfig.use_auxiliary_head
    • EncoderConfig.use_mask_token
    • EncoderConfig.use_mean_pooling
    • EncoderConfig.use_relative_position_bias
    • EncoderConfig.use_shared_relative_position_bias
    • EncoderConfig.vocab_size
  • GenerationConfig
    • GenerationConfig.bos_token_id
    • GenerationConfig.decoder_start_token_id
    • GenerationConfig.early_stopping
    • GenerationConfig.eos_token_id
    • GenerationConfig.length_penalty
    • GenerationConfig.max_length
    • GenerationConfig.no_repeat_ngram_size
    • GenerationConfig.num_beams
    • GenerationConfig.pad_token_id

Module contents