← all repositories

google-research/big_transfer

A repository providing pre-trained Big Transfer (BiT) models for visual representation learning and multi-framework fine-tuning code.

1.5k stars Python Computer VisionML Frameworks
big_transfer
Velocity · 7d
+0.7
★ / day
Trend
steady
star history

This repository releases pre-trained models from the Big Transfer (BiT) paper, trained on ILSVRC-2012 and ImageNet-21k datasets. It provides code to fine-tune these models across TensorFlow 2, PyTorch, and Jax/Flax frameworks. The models enable transfer learning for various computer vision tasks through high-quality visual representations learned via large-scale pre-training on ImageNet data.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.