SuSyGame / config.json
dariog's picture
First commit
34b9ee4
raw
history blame
170 Bytes
{
"model_architecture": "ResNet18",
"num_classes": 6,
"input_size": [224, 224],
"pretrained": true,
"learning_rate": 0.0001,
"batch_size": 256
}