peterhuang24's picture
Upload config.json
46e609b verified
raw
history blame
280 Bytes
{
"model_type": "resnet",
"input_shape": [224, 224, 3],
"num_classes": 2,
"optimizer": {
"name": "Adam",
"learning_rate": 0.001
},
"batch_size": 32,
"epochs": 10,
"loss_function": "binary_crossentropy",
"metrics": ["accuracy"]
}