SAVRN
Search Contact SAVRN

Open-weight model · Image classification

densenet121.ra_in1k

by PyTorch Image Models timm/densenet121.ra_in1k

densenet121.ra_in1k is an open-weight model for image classification from PyTorch Image Models, released under Apache License 2.0. It has 8M parameters. At 16-bit it needs about 0 GB of GPU memory, which fits on 1x MI300X from $1.85 an hour, at the lowest prices in the SAVRN Index. It draws 114.2k downloads a month.

A DenseNet image classification model. Pretrained on ImageNet-1k in timm by Ross Wightman using RandAugment RA recipe. Related to B recipe in ResNet Strikes Back.

Parameters8M
Context—
Weights64.9 MB
Licenseapache-2.0
AccessOpen weights
Monthly Downloads114.2k

Runs On

What it takes to serve densenet121.ra_in1k (8M parameters): the memory its weights need at each precision, and the cheapest way to rent enough data-center GPUs to hold them.

PrecisionWeightsMemory neededCheapest setupPer hourAlso fits
16-bit 0.0 GB 0.0 GB 1x MI300X (192 GB)
Vultr
$1.85 1x H100 $1.99 · 1x MI325X $2.00
8-bit 0.0 GB 0.0 GB 1x MI300X (192 GB)
Vultr
$1.85 1x H100 $1.99 · 1x MI325X $2.00
4-bit 0.0 GB 0.0 GB 1x MI300X (192 GB)
Vultr
$1.85 1x H100 $1.99 · 1x MI325X $2.00

Memory is the weights at that precision plus 20% for the runtime and a short context; a long context needs more. Prices are the lowest on-demand hourly rates in the SAVRN Index, read Oct 9, 2026.

densenet121.ra_in1k on every accelerator the SAVRN Index prices, at every precision

Model Card

By PyTorch Image Models, published under apache-2.0, revision 92007b6200e0.

Model card for densenet121.ra_in1k

A DenseNet image classification model. Pretrained on ImageNet-1k in timm by Ross Wightman using RandAugment RA recipe. Related to B recipe in ResNet Strikes Back.

Model Details

  • Model Type: Image classification / feature backbone
  • Model Stats:
  • Params (M): 8.0
  • GMACs: 2.9
  • Activations (M): 6.9
  • Image size: train = 224 x 224, test = 288 x 288
  • Papers:
  • Densely Connected Convolutional Networks: https://arxiv.org/abs/1608.06993
  • ResNet strikes back: An improved training procedure in timm: https://arxiv.org/abs/2110.00476
  • Dataset: ImageNet-1k
  • Original: https://github.com/huggingface/pytorch-image-models

Model Usage

Image Classification

from urllib.request import urlopen
from PIL import Image
import timm

img = Image.open(urlopen(
    'https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/beignets-task-guide.png'
))

model = timm.create_model('densenet121.ra_in1k', pretrained=True)
model = model.eval()

# get model specific transforms (normalization, resize)
data_config = timm.data.resolve_model_data_config(model)
transforms = timm.data.create_transform(**data_config, is_training=False)

output = model(transforms(img).unsqueeze(0))  # unsqueeze single image into batch of 1

top5_probabilities, top5_class_indices = torch.topk(output.softmax(dim=1) * 100, k=5)

Feature Map Extraction

Read the full model card (384 words)

Identity and Version

Repository
timm/densenet121.ra_in1k
Publisher
PyTorch Image Models
Task
Image classification
Modality
Image
Library
timm
Parameters
8M parameters
Languages
Not stated by the source
Revision
92007b6200e0b4a4fe68cb4e3947022a928aaaae
First published
2023-04-21
Last updated
2025-01-21

Files and Weights

5 files, 64.9 MB in total. The weights are 2 files totalling 64.9 MB in bin, safetensors.

Weights2 files · 64.9 MB
Configuration1 file · 668 B
Documentation1 file · 4.0 KB
Repository1 file · 1.5 KB
Every file
FileTypeSizeSHA-256
model.safetensorsWeights32.3 MB eae900d8ae5b
pytorch_model.binWeights32.5 MB e73d2a763b34
config.jsonConfiguration668 B —
README.mdDocumentation4.0 KB —
.gitattributesRepository1.5 KB —

License and Download

License
apache-2.0
Access
Open weights, no gate
Download size
64.9 MB
Download from PyTorch Image Models

Released by PyTorch Image Models through its official repository on Hugging Face. Read the license.

Built From

  • Described by arXiv:1608.06993
  • Described by arXiv:2110.00476
  • Trained on (disclosed) imagenet-1k

Memory Requirements

PrecisionWeights in memory
As published64.9 MB
16-bit0.0 GB
8-bit0.0 GB
4-bit0.0 GB

Weights only, from the published parameter count; the key-value cache and runtime add to this.

Questions About densenet121.ra_in1k

How much GPU memory does densenet121.ra_in1k need?

About 0 GB at 16-bit and 0 GB at 4-bit: the weights (8M parameters) plus a working margin. A long context needs more.

What is the cheapest GPU to run densenet121.ra_in1k on?

At 16-bit, 1x MI300X from $1.85 an hour; at 4-bit, 1x MI300X from $1.85 an hour, at the lowest on-demand prices the SAVRN Index lists.

Can I use densenet121.ra_in1k commercially?

Yes. densenet121.ra_in1k is released under Apache License 2.0. The Apache License 2.0 is a permissive open-source license. It permits commercial use, modification and redistribution. It requires keeping the license and copyright notices and any NOTICE file, stating significant changes, and it includes an express patent grant from contributors.

Similar Models

Model · Image classification

convnextv2_pico.fcmae_ft_in1k

PyTorch Image Models

A ConvNeXt-V2 image classification model. Pretrained with a fully convolutional masked autoencoder framework (FCMAE) and fine-tuned on ImageNet-1k. Explore the dataset and runtime metrics of this model in timm model results. All timing numbers from eager model PyTorch 1.13 on RTX 3090 w/ AMP.

Open weights cc-by-nc-4.0 9M parameters timm

Model · Image classification

repvgg_a0.rvgg_in1k

PyTorch Image Models

A RepVGG image classification model. Trained on ImageNet-1k by paper authors. This model architecture is implemented using timm's flexible BYOBNet (Bring-Your-Own-Blocks Network). block / stage layout stem layout output stride (dilation) activation and norm layers channel and spatial / self-attention layers...and also includes timm features common to many other architectures, including: stochastic depth gradient checkpointing layer-wise LR decay per-stage feature extraction Explore the dataset and runtime metrics of this model in timm model results.

Open weights mit 9M parameters timm

A Vision Transformer (ViT) image classification model. Trained on ImageNet-21k and fine-tuned on ImageNet-1k (with additional augmentation and regularization) in JAX by paper authors, ported to PyTorch by Ross Wightman. - How to train your ViT? Data, Augmentation, and Regularization in Vision Transformers: https://arxiv.org/abs/2106.10270 - An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale: https://arxiv.org/abs/2010.11929v2 Explore the dataset and runtime metrics of this model in timm model results.

Open weights apache-2.0 6M parameters timm

Model · Image classification

deit_tiny_patch16_224.fb_in1k

PyTorch Image Models

A DeiT image classification model. Trained on ImageNet-1k by paper authors. - Training data-efficient image transformers & distillation through attention: https://arxiv.org/abs/2012.12877 Explore the dataset and runtime metrics of this model in timm model results.

Open weights apache-2.0 6M parameters timm

Model · Image classification

efficientnet_b0.ra_in1k

PyTorch Image Models

A EfficientNet image classification model. Trained on ImageNet-1k in timm using recipe template described below. RandAugment RA recipe. Inspired by and evolved from EfficientNet RandAugment recipes. Published as B recipe in ResNet Strikes Back. RMSProp (TF 1.0 behaviour) optimizer, EMA weight averaging Step (exponential decay w/ staircase) LR schedule with warmup Explore the dataset and runtime metrics of this model in timm model results.

Open weights apache-2.0 5M parameters timm