Why Do Self-Supervised Models Transfer? Investigating the Impact of Invariance on Downstream Tasks
This repository contains the official code for the paper Why Do Self-Supervised Models Transfer? Investigating the Impact of Invariance on Downstream Tasks. Requirements This codebase has been tested with the following package versions: python=3.8.8 torch=1.9.0+cu102 torchvision=0.10.0+cu102 PIL=8.1.0 numpy=1.19.2 scipy=1.6.1 tqdm=4.57.0 sklearn=0.24.1 albumentations=1.0.3 Prepare data There are several classes defined in the datasets directory. The data is expected in a directory name data, located on the same level as this repository. Below is an outline of the expected file structure: data/ […]
Read more