#
TVLT

##
Overview

The TVLT model was proposed in TVLT: Textless Vision-Language Transformer by Zineng Tang, Jaemin Cho, Yixin Nie, Mohit Bansal (the first three authors contributed equally). The Textless Vision-Language Transformer (TVLT) is a model that uses raw visual and audio inputs for vision-and-language representation learning, without using text-specific modules such as tokenization or automatic speech recognition (ASR). It can perform various audiovisual and vision-language tasks like retrieval, question answering, etc.

The abstract from the paper is the following:

In this work, we present the Textless Vision-Language Transformer (TVLT), where homogeneous transformer blocks take raw visual and audio inputs for vision-and-language representation learning with minimal modality-specific design, and do not use text-specific modules such as tokenization or automatic speech recognition (ASR). TVLT is trained by reconstructing masked patches of continuous video frames and audio spectrograms (masked autoencoding) and contrastive modeling to align video and audio. TVLT attains performance comparable to its text-based counterpart on various multimodal tasks, such as visual question answering, image retrieval, video retrieval, and multimodal sentiment analysis, with 28x faster inference speed and only 1/3 of the parameters. Our findings suggest the possibility of learning compact and efficient visual-linguistic representations from low-level visual and audio signals without assuming the prior existence of text.

Tips:

- TVLT is a model that takes both `pixel_values` and `audio_values` as input. One can use TvltProcessor to prepare data for the model. This processor wraps an image processor (for the image/video modality) and an audio feature extractor (for the audio modality) into one.
- TVLT is trained with images/videos and audios of various sizes: the authors resize and crop the input images/videos to 224 and limit the length of audio spectrogram to 2048. To make batching of videos and audios possible, the authors use a `pixel_mask` that indicates which pixels are real/padding and `audio_mask` that indicates which audio values are real/padding.
- The design of TVLT is very similar to that of a standard Vision Transformer (ViT) and masked autoencoder (MAE) as in ViTMAE. The difference is that the model includes embedding layers for the audio modality.
- The PyTorch version of this model is only available in torch 1.10 and higher.

TVLT architecture. Taken from the original paper.

The original code can be found here. This model was contributed by Zineng Tang.

##
TvltConfig

###
class transformers.TvltConfig

< source >

( image_size = 224 spectrogram_length = 2048 frequency_length = 128 image_patch_size = [16, 16] audio_patch_size = [16, 16] num_image_channels = 3 num_audio_channels = 1 num_frames = 8 hidden_size = 768 num_hidden_layers = 12 num_attention_heads = 12 intermediate_size = 3072 hidden_act = 'gelu' hidden_dropout_prob = 0.0 attention_probs_dropout_prob = 0.0 initializer_range = 0.02 layer_norm_eps = 1e-06 qkv_bias = True use_mean_pooling = False decoder_num_attention_heads = 16 decoder_hidden_size = 512 decoder_num_hidden_layers = 8 decoder_intermediate_size = 2048 pixel_mask_ratio = 0.75 audio_mask_ratio = 0.15 audio_mask_type = 'frame-level' task_matching = True task_mae = True loss_type = 'classification' **kwargs )

Parameters

- image_size (`int`, optional, defaults to 224) — The size (resolution) of each image.
- spectrogram_length (`int`, optional, defaults to 2048) — The time length of each audio spectrogram.
- frequency_length (`int`, optional, defaults to 128) — The frequency length of audio spectrogram.
- image_patch_size (`List[int]`, optional, defaults to `[16, 16]`) — The size (resolution) of each image patch.
- audio_patch_size (`List[int]`, optional, defaults to `[16, 16]`) — The size (resolution) of each audio patch.
- num_image_channels (`int`, optional, defaults to 3) — The number of input image channels.
- num_audio_channels (`int`, optional, defaults to 1) — The number of input audio channels.
- num_frames (`int`, optional, defaults to 8) — The maximum number of frames for an input video.
- hidden_size (`int`, optional, defaults to 768) — Dimensionality of the encoder layers and the pooler layer.
- num_hidden_layers (`int`, optional, defaults to 12) — Number of hidden layers in the Transformer encoder.
- num_attention_heads (`int`, optional, defaults to 12) — Number of attention heads for each attention layer in the Transformer encoder.
- intermediate_size (`int`, optional, defaults to 3072) — Dimensionality of the “intermediate” (i.e., feed-forward) layer in the Transformer encoder.
- hidden_act (`str` or `function`, optional, defaults to `"gelu"`) — The non-linear activation function (function or string) in the encoder and pooler. If string, `"gelu"`, `"relu"`, `"selu"` and `"gelu_new"` are supported.
- hidden_dropout_prob (`float`, optional, defaults to 0.0) — The dropout probabilitiy for all fully connected layers in the embeddings, encoder, and pooler.
- attention_probs_dropout_prob (`float`, optional, defaults to 0.0) — The dropout ratio for the attention probabilities.
- initializer_range (`float`, optional, defaults to 0.02) — The standard deviation of the truncated_normal_initializer for initializing all weight matrices.
- layer_norm_eps (`float`, optional, defaults to 1e-6) — The epsilon used by the layer normalization layers.
- qkv_bias (`bool`, optional, defaults to `True`) — Whether to add a bias to the queries, keys and values.
- use_mean_pooling (`bool`, optional, defaults to `False`) — Whether to mean pool the final hidden states instead of using the final hidden state of the [CLS] token.
- decoder_num_attention_heads (`int`, optional, defaults to 16) — Number of attention heads for each attention layer in the decoder.
- decoder_hidden_size (`int`, optional, defaults to 512) — Dimensionality of the decoder.
- decoder_num_hidden_layers (`int`, optional, defaults to 8) — Number of hidden layers in the decoder.
- decoder_intermediate_size (`int`, optional, defaults to 2048) — Dimensionality of the “intermediate” (i.e., feed-forward) layer in the decoder.
- pixel_mask_ratio (`float`, optional, defaults to 0.75) — Image patch masking ratio.
- audio_mask_ratio (`float`, optional, defaults to 0.15) — Audio patch masking ratio.
- audio_mask_type (`str`, optional, defaults to `"frame-level"`) — Audio patch masking type, choose between “frame-level” and “patch-level”.
- task_matching (`bool`, optional, defaults to `True`) — Whether to use vision audio matching task in pretraining.
- task_mae (`bool`, optional, defaults to `True`) — Whether to use the masked auto-encoder (MAE) in pretraining.
- loss_type (`str`, optional, defaults to `"classification"`) — Loss types including regression and classification.

This is the configuration class to store the configuration of a TvltModel. It is used to instantiate a TVLT model according to the specified arguments, defining the model architecture. Instantiating a configuration with the defaults will yield a similar configuration to that of the TVLT ZinengTang/tvlt-base architecture.

Configuration objects inherit from PretrainedConfig and can be used to control the model outputs. Read the documentation from PretrainedConfig for more information.

Example:

    >>> from transformers import TvltConfig, TvltModel

    >>> # # Initializing a TVLT ZinengTang/tvlt-base style configuration
    >>> configuration = TvltConfig()

    >>> # # Initializing a model (with random weights) from the ZinengTang/tvlt-base style configuration
    >>> model = TvltModel(configuration)

    >>> # Accessing the model configuration
    >>> configuration = model.config

##
TvltProcessor

###
class transformers.TvltProcessor

< source >

( image_processor feature_extractor )

Parameters

- image_processor (`TvltImageProcessor`) — An instance of TvltImageProcessor. The image processor is a required input.
- feature_extractor (`TvltFeatureExtractor`) — An instance of TvltFeatureExtractor. The feature extractor is a required input.

Constructs a TVLT processor which wraps a TVLT image processor and TVLT feature extractor into a single processor.

TvltProcessor offers all the functionalities of TvltImageProcessor and TvltFeatureExtractor. See the docstring of call() for more information.

####
__call__

< source >

( images = None audio = None images_mixed = None sampling_rate = None mask_audio = False mask_pixel = False *args **kwargs )

Forwards the `images` argument to TvltImageProcessor’s preprocess() and the `audio` argument to TvltFeatureExtractor’s call(). Please refer to the docstring of the above two methods for more information.

##
TvltImageProcessor

###
class transformers.TvltImageProcessor

< source >

( do_resize: bool = True size: typing.Dict[str, int] = None patch_size: typing.List[int] = [16, 16] num_frames: int = 8 resample: Resampling = <Resampling.BILINEAR: 2> do_center_crop: bool = True crop_size: typing.Dict[str, int] = None do_rescale: bool = True rescale_factor: typing.Union[int, float] = 0.00392156862745098 do_normalize: bool = True image_mean: typing.Union[float, typing.List[float], NoneType] = [0.5, 0.5, 0.5] image_std: typing.Union[float, typing.List[float], NoneType] = [0.5, 0.5, 0.5] init_mask_generator = False **kwargs )

Parameters

- do_resize (`bool`, optional, defaults to `True`) — Whether to resize the image’s (height, width) dimensions to the specified `size`. Can be overridden by the `do_resize` parameter in the `preprocess` method.
- size (`Dict[str, int]` optional, defaults to `{"shortest_edge" -- 224}`): Size of the output image after resizing. The shortest edge of the image will be resized to `size["shortest_edge"]` while maintaining the aspect ratio of the original image. Can be overriden by `size` in the `preprocess` method.
- patch_size (`List[int]` optional, defaults to [16,16]) — The patch size of image patch embedding.
- num_frames (`int` optional, defaults to 8) — The maximum number of video frames.
- resample (`PILImageResampling`, optional, defaults to `PILImageResampling.BILINEAR`) — Resampling filter to use if resizing the image. Can be overridden by the `resample` parameter in the `preprocess` method.
- do_center_crop (`bool`, optional, defaults to `True`) — Whether to center crop the image to the specified `crop_size`. Can be overridden by the `do_center_crop` parameter in the `preprocess` method.
- crop_size (`Dict[str, int]`, optional, defaults to `{"height" -- 224, "width": 224}`): Size of the image after applying the center crop. Can be overridden by the `crop_size` parameter in the `preprocess` method.
- do_rescale (`bool`, optional, defaults to `True`) — Whether to rescale the image by the specified scale `rescale_factor`. Can be overridden by the `do_rescale` parameter in the `preprocess` method.
- rescale_factor (`int` or `float`, optional, defaults to 1/255) — Defines the scale factor to use if rescaling the image. Can be overridden by the `rescale_factor` parameter in the `preprocess` method.
- do_normalize (`bool`, optional, defaults to `True`) — Whether to normalize the image. Can be overridden by the `do_normalize` parameter in the `preprocess` method.
- image_mean (`float` or `List[float]`, optional, defaults to `IMAGENET_STANDARD_MEAN`) — Mean to use if normalizing the image. This is a float or list of floats the length of the number of channels in the image. Can be overridden by the `image_mean` parameter in the `preprocess` method.
- image_std (`float` or `List[float]`, optional, defaults to `IMAGENET_STANDARD_STD`) — Standard deviation to use if normalizing the image. This is a float or list of floats the length of the number of channels in the image. Can be overridden by the `image_std` parameter in the `preprocess` method.

Constructs a TVLT image processor.

This processor can be used to prepare either videos or images for the model by converting images to 1-frame videos.

####
preprocess

< source >

( videos: typing.Union[ForwardRef('PIL.Image.Image'), numpy.ndarray, ForwardRef('torch.Tensor'), typing.List[ForwardRef('PIL.Image.Image')], typing.List[numpy.ndarray], typing.List[ForwardRef('torch.Tensor')]] do_resize: bool = None size: typing.Dict[str, int] = None patch_size: typing.List[int] = None num_frames: int = None resample: Resampling = None do_center_crop: bool = None crop_size: typing.Dict[str, int] = None do_rescale: bool = None rescale_factor: float = None do_normalize: bool = None image_mean: typing.Union[float, typing.List[float], NoneType] = None image_std: typing.Union[float, typing.List[float], NoneType] = None is_mixed: bool = False return_tensors: typing.Union[str, transformers.utils.generic.TensorType, NoneType] = None data_format: ChannelDimension = <ChannelDimension.FIRST: 'channels_first'> **kwargs ) → BatchFeature

Parameters

- videos (`ImageInput`) — Images or videos to preprocess.
- do_resize (`bool`, optional, defaults to `self.do_resize`) — Whether to resize the image.
- size (`Dict[str, int]`, optional, defaults to `self.size`) — Size of the image after applying resize.
- patch_size (`List[int]` optional, defaults to self.patch_size) — The patch size of image patch embedding.
- num_frames (`int` optional, defaults to self.num_frames) — The maximum number of video frames.
- resample (`PILImageResampling`, optional, defaults to `self.resample`) — Resampling filter to use if resizing the image. This can be one of the enum `PILImageResampling`, Only has an effect if `do_resize` is set to `True`.
- do_center_crop (`bool`, optional, defaults to `self.do_centre_crop`) — Whether to centre crop the image.
- crop_size (`Dict[str, int]`, optional, defaults to `self.crop_size`) — Size of the image after applying the centre crop.
- do_rescale (`bool`, optional, defaults to `self.do_rescale`) — Whether to rescale the image values between [0 - 1].
- rescale_factor (`float`, optional, defaults to `self.rescale_factor`) — Rescale factor to rescale the image by if `do_rescale` is set to `True`.
- do_normalize (`bool`, optional, defaults to `self.do_normalize`) — Whether to normalize the image.
- image_mean (`float` or `List[float]`, optional, defaults to `self.image_mean`) — Image mean.
- image_std (`float` or `List[float]`, optional, defaults to `self.image_std`) — Image standard deviation.
- is_mixed (`bool`, optional) — If the input video has negative samples.
- return_tensors (`str` or `TensorType`, optional) — The type of tensors to return. Can be one of:
  - Unset: Return a list of `np.ndarray`.
  - `TensorType.TENSORFLOW` or `'tf'`: Return a batch of type `tf.Tensor`.
  - `TensorType.PYTORCH` or `'pt'`: Return a batch of type `torch.Tensor`.
  - `TensorType.NUMPY` or `'np'`: Return a batch of type `np.ndarray`.
  - `TensorType.JAX` or `'jax'`: Return a batch of type `jax.numpy.ndarray`.
- data_format (`ChannelDimension` or `str`, optional, defaults to `ChannelDimension.FIRST`) — The channel dimension format for the output image. Can be one of:
  - `ChannelDimension.FIRST`: image in (num_channels, height, width) format.
  - `ChannelDimension.LAST`: image in (height, width, num_channels) format.
  - Unset: Use the inferred channel dimension format of the input image.

Returns

BatchFeature

A BatchFeature with the following fields:

- pixel_values — Pixel values to be fed to a model, of shape (batch_size, num_channels, height, width).

- pixel_mask — Pixel masks to be fed to a model, of shape (batch_size, num_pixel_patches).

- pixel_values_mixed — Pixel values with both postive or negative to be fed to a model, of shape (batch_size, num_channels, height, width).

- pixel_mask_mixed — Pixel masks with both postive or negative to be fed to a model, of shape (batch_size, num_pixel_patches).

Preprocess an videos or image or batch of videos or images.

##
TvltFeatureExtractor

###
class transformers.TvltFeatureExtractor

< source >

( spectrogram_length = 2048 num_channels = 1 patch_size = [16, 16] feature_size = 128 sampling_rate = 44100 hop_length_to_sampling_rate = 86 n_fft = 2048 padding_value = 0.0 **kwargs )

Parameters

- spectrogram_length (`Dict[str, int]` optional, defaults to 2048) — The time length of each audio spectrogram.
- num_channels (`int` optional, defaults to 1) — Number of audio channels.
- patch_size (`List[int]` optional, defaults to `[16, 16]`) — The patch size of audio patch embedding.
- feature_size (`int`, defaults to 128) — The frequency length of audio spectrogram.
- sampling_rate (`int`, defaults to 44100) — The sampling rate at which the audio files should be digitalized expressed in Hertz (Hz).
- hop_length_to_sampling_rate (`int`, defaults to 86) — Hop length is length of the overlaping windows for the STFT used to obtain the Mel Frequency coefficients. For example, with sampling rate 44100, the hop length is 512, with 44100 / 512 = 86
- n_fft (`int`, defaults to 2048) — Size of the Fourier transform.
- padding_value (`float`, optional, defaults to 0.0) — Padding value used to pad the audio. Should correspond to silences.

Constructs a TVLT audio feature extractor. This feature extractor can be used to prepare audios for the model.

This feature extractor inherits from FeatureExtractionMixin which contains most of the main methods. Users should refer to this superclass for more information regarding those methods.

####
__call__

< source >

( raw_speech: typing.Union[numpy.ndarray, typing.List[float], typing.List[numpy.ndarray], typing.List[typing.List[float]]] return_tensors: typing.Union[str, transformers.utils.generic.TensorType, NoneType] = None return_attention_mask: typing.Optional[bool] = True sampling_rate: typing.Optional[int] = None resample: bool = False mask_audio: bool = False **kwargs ) → BatchFeature

Parameters

- raw_speech (`np.ndarray`, `List[float]`, `List[np.ndarray]`, `List[List[float]]`) — The sequence or batch of sequences to be padded. Each sequence can be a numpy array, a list of float values, a list of numpy arrays or a list of list of float values. Must be mono channel audio, not stereo, i.e. single float per timestep.
- return_tensors (`str` or TensorType, optional) — If set, will return tensors instead of list of python integers. Acceptable values are:
  - `'pt'`: Return PyTorch `torch.Tensor` objects.
  - `'np'`: Return Numpy `np.ndarray` objects.
- return_attention_mask (`bool`, optional, default to `True`) — Whether to return the attention mask. If left to the default, will return the attention mask according to the specific feature_extractor’s default. What are attention masks?

  For TvltTransformer models, `attention_mask` should alwys be passed for batched inference, to avoid subtle bugs.

- sampling_rate (`int`, optional) — The sampling rate at which the `raw_speech` input was sampled. It is strongly recommended to pass `sampling_rate` at the forward call to prevent silent errors and allow automatic speech recognition pipeline. Current model supports sampling rate 16000 and 44100.
- resample (`bool`, optional, defaults to `False`) — If the sampling rate is not matched, resample the input audio to match.
- mask_audio (`bool`, optional, defaults to `False`) — Whether or not to mask input audio for MAE task.

Returns

BatchFeature

A BatchFeature with the following fields:

- audio_values — Audio values to be fed to a model, of shape (batch_size, num_channels, height, width).

- audio_mask — Audio masks to be fed to a model, of shape (batch_size, num_audio_patches).

Main method to prepare one or several audio(s) for the model.

##
TvltModel

###
class transformers.TvltModel

< source >

( config )

Parameters

- config (TvltConfig) — Model configuration class with all the parameters of the model. Initializing with a config file does not load the weights associated with the model, only the configuration. Check out the from_pretrained() method to load the model weights.

The bare TVLT Model transformer outputting raw hidden-states without any specific head on top. This model is a PyTorch torch.nn.Module subclass. Use it as a regular PyTorch Module and refer to the PyTorch documentation for all matter related to general usage and behavior.

####
forward

< source >

( pixel_values audio_values pixel_mask = None audio_mask = None mask_pixel = False mask_audio = False output_attentions = None output_hidden_states = None return_dict = None ) → `transformers.models.tvlt.modeling_tvlt.TvltModelOutput` or `tuple(torch.FloatTensor)`

Parameters

- pixel_values (`torch.FloatTensor` of shape `(batch_size, num_frames, num_channels, height, width)`) — Pixel values. Pixel values can be obtained using TvltProcessor. See TvltProcessor.call() for details.
- audio_values (`torch.FloatTensor` of shape `(batch_size, num_channels, height, width)`) — Audio values. Audio values can be obtained using TvltProcessor. See TvltProcessor.call() for details.
- pixel_mask (`torch.FloatTensor` of shape `(batch_size, num_pixel_patches)`) — Pixel masks. Pixel masks can be obtained using TvltProcessor. See TvltProcessor.call() for details.
- audio_mask (`torch.FloatTensor` of shape `(batch_size, num_audio_patches)`) — Audio masks. Audio masks can be obtained using TvltProcessor. See TvltProcessor.call() for details.
- pixel_values_mixed (`torch.FloatTensor` of shape `(batch_size, num_frames, num_channels, height, width)`) — Pixel values that mix positive and negative samples in Tvlt vision-audio matching. Pixel values mixed can be obtained using TvltProcessor. See TvltProcessor.call() for details.
- pixel_mask_mixed (`torch.FloatTensor` of shape `(batch_size, num_channels, height, width)`) — Pixel masks of pixel_values_mixed. Pixel masks mixed can be obtained using TvltProcessor. See TvltProcessor.call() for details.
- mask_pixel (`bool`, optional) — Whether to mask pixel for MAE tasks. Only set to True in TvltForPreTraining.
- mask_audio (`bool`, optional) — Whether to mask audio for MAE tasks. Only set to True in TvltForPreTraining.
- output_attentions (`bool`, optional) — Whether or not to return the attentions tensors of all attention layers. See `attentions` under returned tensors for more detail.
- output_hidden_states (`bool`, optional) — Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for more detail.
- return_dict (`bool`, optional) — Whether or not to return a ModelOutput instead of a plain tuple.

Returns

`transformers.models.tvlt.modeling_tvlt.TvltModelOutput` or `tuple(torch.FloatTensor)`

A `transformers.models.tvlt.modeling_tvlt.TvltModelOutput` or a tuple of `torch.FloatTensor` (if `return_dict=False` is passed or when `config.return_dict=False`) comprising various elements depending on the configuration (TvltConfig) and inputs.

- last_hidden_state (`torch.FloatTensor` of shape `(batch_size, sequence_length, hidden_size)`) — Sequence of hidden-states at the output of the last layer of the model.
- last_pixel_hidden_state (`torch.FloatTensor` of shape `(batch_size, pixel_sequence_length, hidden_size)`) — Pixel sequence of hidden-states at the output of the last layer of the model.
- last_audio_hidden_state (`torch.FloatTensor` of shape `(batch_size, audio_sequence_length, hidden_size)`) — Audio sequence of hidden-states at the output of the last layer of the model.
- pixel_label_masks (`torch.FloatTensor` of shape `(batch_size, pixel_patch_length)`) — Tensor indicating which pixel patches are masked (1) and which are not (0).
- audio_label_masks (`torch.FloatTensor` of shape `(batch_size, audio_patch_length)`) — Tensor indicating which audio patches are masked (1) and which are not (0).
- pixel_ids_restore (`torch.LongTensor` of shape `(batch_size, pixel_patch_length)`) — Tensor containing the ids permutation of pixel masking.
- audio_ids_restore (`torch.LongTensor` of shape `(batch_size, audio_patch_length)`) — Tensor containing the ids permutation of audio masking.
- hidden_states (`tuple(torch.FloatTensor)`, optional, returned when `output_hidden_states=True` is passed or when `config.output_hidden_states=True`) — Tuple of `torch.FloatTensor` (one for the output of the embeddings and one for the output of each layer) of shape `(batch_size, sequence_length, hidden_size)`. Hidden-states of the model at the output of each layer plus the initial embedding outputs.
- attentions (`tuple(torch.FloatTensor)`, optional, returned when `output_attentions=True` is passed or when `config.output_attentions=True`) — Tuple of `torch.FloatTensor` (one for each layer) of shape `(batch_size, num_heads, sequence_length, sequence_length)`. Attentions weights after the attention softmax, used to compute the weighted average in the self-attention heads.

The TvltModel forward method, overrides the `__call__` special method.

Although the recipe for forward pass needs to be defined within this function, one should call the `Module` instance afterwards instead of this since the former takes care of running the pre and post processing steps while the latter silently ignores them.

Examples:

    >>> from transformers import TvltProcessor, TvltModel
    >>> import numpy as np
    >>> import torch

    >>> num_frames = 8
    >>> images = list(np.random.randn(num_frames, 3, 224, 224))
    >>> audio = list(np.random.randn(10000))

    >>> processor = TvltProcessor.from_pretrained("ZinengTang/tvlt-base")
    >>> model = TvltModel.from_pretrained("ZinengTang/tvlt-base")

    >>> input_dict = processor(images, audio, sampling_rate=44100, return_tensors="pt")

    >>> outputs = model(**input_dict)
    >>> loss = outputs.loss

##
TvltForPreTraining

###
class transformers.TvltForPreTraining

< source >

( config )

Parameters

- config (TvltConfig) — Model configuration class with all the parameters of the model. Initializing with a config file does not load the weights associated with the model, only the configuration. Check out the from_pretrained() method to load the model weights.

The TVLT Model transformer with the decoder on top for self-supervised pre-training. This model is a PyTorch torch.nn.Module subclass. Use it as a regular PyTorch Module and refer to the PyTorch documentation for all matter related to general usage and behavior.

####
forward

< source >

( pixel_values audio_values pixel_mask = None audio_mask = None labels = None pixel_values_mixed = None pixel_mask_mixed = None output_attentions = None output_hidden_states = None return_dict = None ) → `transformers.models.tvlt.modeling_tvlt.TvltForPreTrainingOutput` or `tuple(torch.FloatTensor)`

Parameters

- pixel_values (`torch.FloatTensor` of shape `(batch_size, num_frames, num_channels, height, width)`) — Pixel values. Pixel values can be obtained using TvltProcessor. See TvltProcessor.call() for details.
- audio_values (`torch.FloatTensor` of shape `(batch_size, num_channels, height, width)`) — Audio values. Audio values can be obtained using TvltProcessor. See TvltProcessor.call() for details.
- pixel_mask (`torch.FloatTensor` of shape `(batch_size, num_pixel_patches)`) — Pixel masks. Pixel masks can be obtained using TvltProcessor. See TvltProcessor.call() for details.
- audio_mask (`torch.FloatTensor` of shape `(batch_size, num_audio_patches)`) — Audio masks. Audio masks can be obtained using TvltProcessor. See TvltProcessor.call() for details.
- pixel_values_mixed (`torch.FloatTensor` of shape `(batch_size, num_frames, num_channels, height, width)`) — Pixel values that mix positive and negative samples in Tvlt vision-audio matching. Pixel values mixed can be obtained using TvltProcessor. See TvltProcessor.call() for details.
- pixel_mask_mixed (`torch.FloatTensor` of shape `(batch_size, num_channels, height, width)`) — Pixel masks of pixel_values_mixed. Pixel masks mixed can be obtained using TvltProcessor. See TvltProcessor.call() for details.
- mask_pixel (`bool`, optional) — Whether to mask pixel for MAE tasks. Only set to True in TvltForPreTraining.
- mask_audio (`bool`, optional) — Whether to mask audio for MAE tasks. Only set to True in TvltForPreTraining.
- output_attentions (`bool`, optional) — Whether or not to return the attentions tensors of all attention layers. See `attentions` under returned tensors for more detail.
- output_hidden_states (`bool`, optional) — Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for more detail.
- return_dict (`bool`, optional) — Whether or not to return a ModelOutput instead of a plain tuple.
- pixel_values_mixed (`torch.FloatTensor` of shape `(batch_size, num_frames, num_channels, height, width)`) — Pixel values that mix positive and negative samples in Tvlt vision-audio matching. Audio values can be obtained using TvltProcessor. See TvltProcessor.call() for details.
- pixel_mask_mixed (`torch.FloatTensor` of shape `(batch_size, num_channels, height, width)`) — Pixel masks of pixel_values_mixed. Pixel values mixed can be obtained using TvltProcessor. See TvltProcessor.call() for details.
- labels (`torch.LongTensor` of shape `(batch_size, num_labels)`, optional) — Labels for computing the vision audio matching loss. Indices should be in `[0, 1]`. num_labels has to be 1.

Returns

`transformers.models.tvlt.modeling_tvlt.TvltForPreTrainingOutput` or `tuple(torch.FloatTensor)`

A `transformers.models.tvlt.modeling_tvlt.TvltForPreTrainingOutput` or a tuple of `torch.FloatTensor` (if `return_dict=False` is passed or when `config.return_dict=False`) comprising various elements depending on the configuration (TvltConfig) and inputs.

- loss (`torch.FloatTensor` of shape `(1,)`) — Pixel reconstruction loss.
- matching_logits (`torch.FloatTensor` of shape `(batch_size, 1)`) — Matching objective logits.
- pixel_logits (`torch.FloatTensor` of shape `(batch_size, pixel_patch_length, image_patch_size ** 3 * pixel_num_channels)`): Pixel reconstruction logits.
- audio_logits (`torch.FloatTensor` of shape `(batch_size, audio_patch_length, image_patch_size[0] * image_patch_size[1])`): Audio reconstruction logits.
- hidden_states (`tuple(torch.FloatTensor)`, optional, returned when `output_hidden_states=True` is passed or when `config.output_hidden_states=True`) — Tuple of `torch.FloatTensor` (one for the output of the embeddings and one for the output of each layer) of shape `(batch_size, sequence_length, hidden_size)`. Hidden-states of the model at the output of each layer plus the initial embedding outputs.
- attentions (`tuple(torch.FloatTensor)`, optional, returned when `output_attentions=True` is passed or when `config.output_attentions=True`) — Tuple of `torch.FloatTensor` (one for each layer) of shape `(batch_size, num_heads, sequence_length, sequence_length)`. Attentions weights after the attention softmax, used to compute the weighted average in the self-attention heads.

The TvltForPreTraining forward method, overrides the `__call__` special method.

Although the recipe for forward pass needs to be defined within this function, one should call the `Module` instance afterwards instead of this since the former takes care of running the pre and post processing steps while the latter silently ignores them.

Examples:

    >>> from transformers import TvltProcessor, TvltForPreTraining
    >>> import numpy as np
    >>> import torch

    >>> num_frames = 8
    >>> images = list(np.random.randn(num_frames, 3, 224, 224))
    >>> images_mixed = list(np.random.randn(num_frames, 3, 224, 224))
    >>> audio = list(np.random.randn(10000))
    >>> processor = TvltProcessor.from_pretrained("ZinengTang/tvlt-base")
    >>> model = TvltForPreTraining.from_pretrained("ZinengTang/tvlt-base")
    >>> input_dict = processor(
    ...     images, audio, images_mixed, sampling_rate=44100, mask_pixel=True, mask_audio=True, return_tensors="pt"
    ... )

    >>> outputs = model(**input_dict)
    >>> loss = outputs.loss

##
TvltForAudioVisualClassification

###
class transformers.TvltForAudioVisualClassification

< source >

( config )

Parameters

- config (TvltConfig) — Model configuration class with all the parameters of the model. Initializing with a config file does not load the weights associated with the model, only the configuration. Check out the from_pretrained() method to load the model weights.

Tvlt Model transformer with a classifier head on top (an MLP on top of the final hidden state of the [CLS] token) for audiovisual classification tasks, e.g. CMU-MOSEI Sentiment Analysis and Audio to Video Retrieval.

This model is a PyTorch torch.nn.Module subclass. Use it as a regular PyTorch Module and refer to the PyTorch documentation for all matter related to general usage and behavior.

####
forward

< source >

( pixel_values audio_values pixel_mask = None audio_mask = None output_attentions = None output_hidden_states = None return_dict = None labels = None ) → transformers.modeling_outputs.SequenceClassifierOutput or `tuple(torch.FloatTensor)`

Parameters

- pixel_values (`torch.FloatTensor` of shape `(batch_size, num_frames, num_channels, height, width)`) — Pixel values. Pixel values can be obtained using TvltProcessor. See TvltProcessor.call() for details.
- audio_values (`torch.FloatTensor` of shape `(batch_size, num_channels, height, width)`) — Audio values. Audio values can be obtained using TvltProcessor. See TvltProcessor.call() for details.
- pixel_mask (`torch.FloatTensor` of shape `(batch_size, num_pixel_patches)`) — Pixel masks. Pixel masks can be obtained using TvltProcessor. See TvltProcessor.call() for details.
- audio_mask (`torch.FloatTensor` of shape `(batch_size, num_audio_patches)`) — Audio masks. Audio masks can be obtained using TvltProcessor. See TvltProcessor.call() for details.
- pixel_values_mixed (`torch.FloatTensor` of shape `(batch_size, num_frames, num_channels, height, width)`) — Pixel values that mix positive and negative samples in Tvlt vision-audio matching. Pixel values mixed can be obtained using TvltProcessor. See TvltProcessor.call() for details.
- pixel_mask_mixed (`torch.FloatTensor` of shape `(batch_size, num_channels, height, width)`) — Pixel masks of pixel_values_mixed. Pixel masks mixed can be obtained using TvltProcessor. See TvltProcessor.call() for details.
- mask_pixel (`bool`, optional) — Whether to mask pixel for MAE tasks. Only set to True in TvltForPreTraining.
- mask_audio (`bool`, optional) — Whether to mask audio for MAE tasks. Only set to True in TvltForPreTraining.
- output_attentions (`bool`, optional) — Whether or not to return the attentions tensors of all attention layers. See `attentions` under returned tensors for more detail.
- output_hidden_states (`bool`, optional) — Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for more detail.
- return_dict (`bool`, optional) — Whether or not to return a ModelOutput instead of a plain tuple.
- labels (`torch.LongTensor` of shape `(batch_size, num_labels)`, optional) — Labels for computing the audiovisual loss. Indices should be in `[0, ..., num_classes-1]` where num_classes refers to the number of classes in audiovisual tasks.

Returns

transformers.modeling_outputs.SequenceClassifierOutput or `tuple(torch.FloatTensor)`

A transformers.modeling_outputs.SequenceClassifierOutput or a tuple of `torch.FloatTensor` (if `return_dict=False` is passed or when `config.return_dict=False`) comprising various elements depending on the configuration (TvltConfig) and inputs.

- loss (`torch.FloatTensor` of shape `(1,)`, optional, returned when `labels` is provided) — Classification (or regression if config.num_labels==1) loss.

- logits (`torch.FloatTensor` of shape `(batch_size, config.num_labels)`) — Classification (or regression if config.num_labels==1) scores (before SoftMax).

- hidden_states (`tuple(torch.FloatTensor)`, optional, returned when `output_hidden_states=True` is passed or when `config.output_hidden_states=True`) — Tuple of `torch.FloatTensor` (one for the output of the embeddings, if the model has an embedding layer, + one for the output of each layer) of shape `(batch_size, sequence_length, hidden_size)`.

  Hidden-states of the model at the output of each layer plus the optional initial embedding outputs.

- attentions (`tuple(torch.FloatTensor)`, optional, returned when `output_attentions=True` is passed or when `config.output_attentions=True`) — Tuple of `torch.FloatTensor` (one for each layer) of shape `(batch_size, num_heads, sequence_length, sequence_length)`.

  Attentions weights after the attention softmax, used to compute the weighted average in the self-attention heads.

The TvltForAudioVisualClassification forward method, overrides the `__call__` special method.

Although the recipe for forward pass needs to be defined within this function, one should call the `Module` instance afterwards instead of this since the former takes care of running the pre and post processing steps while the latter silently ignores them.

Examples:

    >>> from transformers import TvltProcessor, TvltForAudioVisualClassification
    >>> import numpy as np
    >>> import torch

    >>> num_frames = 8
    >>> images = list(np.random.randn(num_frames, 3, 224, 224))
    >>> audio = list(np.random.randn(10000))
    >>> processor = TvltProcessor.from_pretrained("ZinengTang/tvlt-base")
    >>> model = TvltForAudioVisualClassification.from_pretrained("ZinengTang/tvlt-base")
    >>> input_dict = processor(images, audio, sampling_rate=44100, return_tensors="pt")

    >>> outputs = model(**input_dict)
    >>> loss = outputs.loss