STomoya's picture
Add model
883acf3
{
"architecture": "efficientnet_b4",
"num_classes": 1000,
"num_features": 1792,
"pretrained_cfg": {
"tag": "st_safebooru_1k",
"custom_load": true,
"input_size": [
3,
320,
320
],
"test_input_size": [
3,
384,
384
],
"fixed_input_size": false,
"interpolation": "bicubic",
"crop_pct": 1.0,
"crop_mode": "center",
"mean": [
0.6886,
0.6415,
0.6443
],
"std": [
0.183,
0.1852,
0.1849
],
"num_classes": 1000,
"pool_size": [
10,
10
],
"first_conv": "conv_stem",
"classifier": "classifier"
}
}