├── .gitignore ├── README.md ├── requirements.txt ├── simclr ├── __init__.py ├── data.py ├── loss.py ├── mixup_model.py └── model.py └── train.py /.gitignore: -------------------------------------------------------------------------------- 1 | lightning_logs/ 2 | *.log 3 | simclr/__pycache__/ 4 | .vscode/ 5 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dthiagarajan/simclr_pytorch/HEAD/README.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dthiagarajan/simclr_pytorch/HEAD/requirements.txt -------------------------------------------------------------------------------- /simclr/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /simclr/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dthiagarajan/simclr_pytorch/HEAD/simclr/data.py -------------------------------------------------------------------------------- /simclr/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dthiagarajan/simclr_pytorch/HEAD/simclr/loss.py -------------------------------------------------------------------------------- /simclr/mixup_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dthiagarajan/simclr_pytorch/HEAD/simclr/mixup_model.py -------------------------------------------------------------------------------- /simclr/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dthiagarajan/simclr_pytorch/HEAD/simclr/model.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dthiagarajan/simclr_pytorch/HEAD/train.py --------------------------------------------------------------------------------