← all repositories

facebookresearch/moco-v3

PyTorch implementation of MoCo v3 for self-supervised training of ResNet and ViT models on ImageNet.

1.3k stars Python ML FrameworksComputer Vision
moco-v3
Velocity · 7d
+0.7
★ / day
Trend
steady
star history

This repository re-implements the MoCo v3 self-supervised contrastive learning algorithm in PyTorch for GPU training. It supports pre-training ResNet and Vision Transformer architectures on ImageNet-1k using various batch sizes, followed by linear evaluation or end-to-end fine-tuning. The implementation reproduces the results from the original TensorFlow/TPU-based paper.

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