Skip to content

Reproduction of Momentum Contrast for Unsupervised Visual Representation Learning

License

Notifications You must be signed in to change notification settings

peisuke/MomentumContrast.pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

18 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

image

About

Reproduction of Momentum Contrast for Unsupervised Visual Representation Learning

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages