A list-like object whose items are serialized and stored in a torch tensor. When launching a process that uses TorchSerializedList with "fork

Computes and stores the average and current value

HuggingFace tokenizer wrapper

Resize and Keep Ratio Copy & paste from `timm`

Crops the given image at the center. If the image is torch Tensor, it is expected to have [..., H, W] shape, where ... means an arbitrary numb

Parser for extracting captions from the Flickr8k dataset web page.

HuggingFace model adapter

A ResNet class that is similar to torchvision's but contains the following changes: - There are now 3 "stem" convolutions as opposed to 1, wi

https://arxiv.org/abs/2212.00794

An iterable dataset yielding a list of urls.

Sigmoid Loss for Language Image Pre-Training (SigLIP) - https://arxiv.org/abs/2303.15343 @article{zhai2023sigmoid, title={Sigmoid loss for

timm model adapter

Apply Color Jitter to the PIL image with a specified probability.

Apply Gray Scale to the PIL image with a specified probability.

ref: https://arxiv.org/abs/2003.13911 official pytorch codes: https://github.com/tjddus9597/Proxy-Anchor-CVPR2020

CLS token pooling NOTE: this is equivalent to ClsPooler above with use_pooler_output=False

CLS token pooling

`MS Coco Captions <https://cocodataset.org/#captions-2015>`_ Dataset. It requires the `COCO API to be installed <https://github.com/pdollar/coco/

`MS Coco Detection <https://cocodataset.org/#detection-2016>`_ Dataset. It requires the `COCO API to be installed <https://github.com/pdollar/coc

`Flickr30k Entities <https://bryanplummer.com/Flickr30kEntities/>`_ Dataset. Args: root (string): Root directory where images are downloa

`Flickr8k Entities <http://hockenmaier.cs.illinois.edu/8k-pictures.html>`_ Dataset. Args: root (string): Root directory where images are

Subclass torch's LayerNorm (with cast back to input dtype).

Subclass torch's LayerNorm to handle fp16 (by casting to float32 and back).

Max pooling

Mean pooling

HuggingFace tokenizer wrapper for SigLIP T5 compatible sentencepiece vocabs