Hugging Face's logo

PyTorch Image Models

https://github.com/rwightman/pytorch-image-models
Request to join this org

AI & ML interests

Computer Vision

Organization Card
Community About org cards

timm

image/png

PyTorch Image Models (timm) is a collection of image models, layers, utilities, optimizers, schedulers, data-loaders / augmentations, and reference training / validation scripts that aim to pull together a wide variety of SOTA models with ability to reproduce ImageNet training results.

Collections 15

Searching for Better ViT Baselines
Exploring ViT hparams and model shapes for the GPU poor (between tiny and base).

spaces 3

models 1415