mirrors
/
SimCLR
mirror of https://github.com/sthalles/SimCLR.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update README.md

pull/5/head
Thalles Silva 2020-03-11 07:54:24 -03:00 committed by GitHub
parent 07dbb5c706
commit 8637514ee3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
README.md
View File

@ -25,6 +25,8 @@ base_convnet: "resnet18" # The ConvNet base model. Choose one of: "resnet18 or r
use_cosine_similarity: True # Distance metric for contrastive loss. If False, uses dot product
epochs: 40 # Number of epochs to train
num_workers: 4 # Number of workers for the data loader
valid_size: 0.05 # validation set size
eval_every_n_epochs: 2 # frequency to eval the feature representations' quality
```
## Feature Evaluation
@ -36,8 +38,8 @@ Check the ```feature_eval/linear_feature_eval.ipynb``` notebook for reproducebil
|:-------------------:|:------------:|:------------:|:-----:|
| Logistic Regression | PCA Features | - | 36.0% |
| KNN | PCA Features | - | 31.8% |
| Logistic Regression | SimCLR | ResNet-18 | 71.8% |
| KNN | SimCLR | ResNet-18 | 66.7% |
| Logistic Regression | SimCLR | ResNet-18 | 75.0% |
| KNN | SimCLR | ResNet-18 | 70.0% |
## Download pre-trained model