Skip to content

SimCLR weights (ResNet50, 256 batch size, 100 epochs)

Latest
Compare
Choose a tag to compare
@Spijkervet Spijkervet released this 12 Mar 23:00
· 68 commits to master since this release

config.yaml

{
  "batch_size": 256,
  "dataset": "STL10",
  "epochs": 100,
  "fp16": false,
  "fp16_opt_level": "O2",
  "logistic_batch_size": 256,
  "logistic_epochs": 100,
  "model_num": 40,
  "model_path": "logs/0",
  "normalize": true,
  "optimizer": "Adam",
  "projection_dim": 64,
  "resnet": "resnet50",
  "seed": 42,
  "start_epoch": 0,
  "temperature": 0.5,
  "workers": 16
}