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
}