cs-giung's picture
Create config.json
c2d841f verified
{
"_name_or_path": "cs-giung/convnext-v1-small-imagenet21k",
"architectures": [
"ConvNextForImageClassification"
],
"depths": [3, 3, 27, 3],
"drop_path_rate": 0.0,
"hidden_act": "gelu",
"hidden_sizes": [96, 192, 384, 768],
"image_size": 224,
"initializer_range": 0.02,
"layer_norm_eps": 1e-06,
"layer_scale_init_value": 1e-06,
"model_type": "convnext",
"num_channels": 3,
"num_labels": 1000,
"num_stages": 4,
"patch_size": 4,
"torch_dtype": "float32",
"transformers_version": "4.41.1"
}