/miniselfsup

A minimal implementation of self-supervised learning frameworks.

Primary LanguagePythonApache License 2.0Apache-2.0

miniSelfSup

This repository is an unofficial Pytorch implementation of recent state-of-the-art SSL methods.

Installation

Requirements

  • Python >= 3.6, CUDA >= 10.1
  • PyTorch >= 1.8.0, torchvision >= 0.8.1

The codebase is developed using Python 3.6 with PyTorch 1.8.1.

First, clone the repository locally:

git clone https://github.com/weivision/miniselfsup.git

Then, install PyTorch and torchvision:

cd miniselfsup
pip install -r requirements.txt