AM-RADIO: Reduce All Domains Into One

Pretrained Models

Refer to model_results.csv for model versions and their metrics.

HuggingFace Hub

Coming Soon

TorchHub

import torch

# If you don't supply the `version` parameter, the latest ViT version will be returned.
model = torch.hub.load('NVlabs/RADIO', 'radio_model', version='radio_v1', progress=True)

print(model)

Training

Coming Soon