This repository is an unofficial Pytorch implementation of recent state-of-the-art SSL methods.
- 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