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

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.