Skip to content

SimCLR weights (40 epochs)

Compare
Choose a tag to compare
@Spijkervet Spijkervet released this 11 Mar 19:56
· 106 commits to master since this release

A pre-trained SimCLR model with the following parameters:
"batch_size": 256,
"epochs": 40,
"n_out": 64,
"normalize": true,
"resnet": "resnet18",
"seed": 634715003,
"start_epoch": 0,
"temperature": 0.5,
"workers": 16

Accuracy with a logistic regression classifier trained on top of SimCLR on STL-10 test set: 0.72