staged-image-classifier / convnext_config.json
Nightfury16's picture
fixed convnextv2 bug
5050066
raw
history blame contribute delete
521 Bytes
{
"architectures": [
"ConvNextV2ForImageClassification"
],
"depths": [
3,
3,
9,
3
],
"drop_path_rate": 0.1,
"hidden_act": "gelu",
"hidden_sizes": [
96,
192,
384,
768
],
"image_size": 224,
"initializer_range": 0.02,
"layer_norm_eps": 1e-06,
"model_type": "convnextv2",
"num_channels": 3,
"num_stages": 4,
"patch_size": 4,
"torch_dtype": "float32",
"transformers_version": "4.35.2"
}