peisuke/MomentumContrast.pytorch

Reproduction of Momentum Contrast for Unsupervised Visual Representation Learning

Python

122

18 commits

updated Jul 25, 2024

See the code

README

Momentum Contrast

This repository is reproduced code of "Momentum Contrast for Unsupervised Visual Representation Learning"

The paper link is here

How to use

Run the train script.

python train.py

Visualizing the features with t-SNE

The below figure visualizes the extracted features of MNIST with t-SNE.

python test.py

Significant stargazers

Yang An

520 followers · starred Mar 2020

Evgenii Zheltonozhskii

171 followers · starred Jan 2020

Similar repos

sthalles/SimCLR

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual…

2,495

HobbitLong/PyContrast

PyTorch implementation of Contrastive Learning methods

1,992

HobbitLong/CMC

[arXiv 2019] "Contrastive Multiview Coding", also contains implementations for MoCo and InstDis

1,338

HobbitLong/SupContrast

PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)

3,449

lucidrains/byol-pytorch

Usable Implementation of "Bootstrap Your Own Latent" self-supervised learning, from Deepmind, in…

1,900

facebookresearch/deepcluster

Deep Clustering for Unsupervised Learning of Visual Features

1,740

asheeshcric/awesome-contrastive-self-supervised-learning

A comprehensive list of awesome contrastive self-supervised learning papers.

1,310

nakosung/VQ-VAE

VQ-VAE implementation / pytorch

184