promotion_model / config.json
011nuyha's picture
Rename model/config.json to config.json
a2a4855 verified
{
"model_type": "efficientnet",
"model_name": "efficientnet_b0",
"input_size": [224, 224, 3],
"num_classes": 2,
"dropout_rate": 0.2,
"batch_norm_eps": 1e-05,
"batch_norm_momentum": 0.1,
"drop_connect_rate": 0.2,
"depth_multiplier": 1.0,
"width_multiplier": 1.0,
"classifier_dropout": 0.2
}