mirrors/SimCLR
1
0
Fork 0
mirror of https://github.com/sthalles/SimCLR.git synced 2025-06-03 15:03:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
Thalles Silva 2021-01-18 07:39:39 -03:00 committed by GitHub
parent ca4016f50c
commit e3065f3333
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -26,7 +26,7 @@ $ python run.py -data ./datasets --dataset-name stl10 --log-every-n-steps 100 --
If you want to run it on CPU (for debugging purposes) use the ```--disable-cuda``` option.
For 16-bit precision GPU training, make sure to install [!NVIDIA apex](https://github.com/NVIDIA/apex) and used the ```--fp16_precision``` flag.
For 16-bit precision GPU training, make sure to install [NVIDIA apex](https://github.com/NVIDIA/apex) and use the ```--fp16_precision``` flag.
## Feature Evaluation