A PyTorch Google Colab implementation of SimCLR based on ICML 2020 paper A Simple Framework for Contrastive Learning of Visual Representations.
- timm
- kaggle
- opencv-python-headless==4.1.2.30
- wandb
This code is implemented using Google Colab Free GPU and "Apparel images dataset" is used in this implementation.
There are some difference between this implementation and official implementation, the model ('ResNet18','efficientnet_b0') is trained on one COLAB GPU:
Rakib Hossain Rifat
Machine Learning Engineer
Polyfins Technology Inc.