sthalles / SimCLR

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations
https://sthalles.github.io/simple-self-supervised-learning/
MIT License
2.19k stars 458 forks source link

Update simclr.py #30

Open towzeur opened 3 years ago

towzeur commented 3 years ago

fixed logging info 'training with gpu'