CCAiM / config.json
serbekun's picture
Update config.json
0be2bc9 verified
raw
history blame contribute delete
333 Bytes
{
"model_type": "ccaim",
"architectures": ["CCAiMModel"],
"num_parameters": 9907530,
"num_classes": 10,
"width_mult": 1,
"base_channels": [64, 128, 256, 512, 512],
"bottleneck_channels": 1024,
"hidden_act": "relu",
"torch_dtype": "float32",
"feature_size": 1024,
"image_size": 224,
"classifier_dropout": 0.0
}