SimCLR

Implementation of SimCLR (A Simple Framework for Contrastive Learning of Visual Representations)

• Requirements

  • numpy
  • torch
  • torchvision
  • opencv-python

• command

  • python3 main.py --epochs [epochs] --batch_size [B] --temperature [T] --out_dim [out_dim] --num_worker [N] --valid_size [val_size]

Reference

1. A Simple Framework for Contrastive Learning of Visual Representations (https://arxiv.org/abs/2002.05709)
2. STL-10 Dataset (https://cs.stanford.edu/~acoates/stl10/)