sergiocannata
/

cvt-21-finetuned-brs2

Image Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

cvt-21-finetuned-brs2 / config.json

sergiocannata's picture

Training in progress, step 100

47e306f over 1 year ago

raw history blame contribute delete

No virus

1.38 kB

	{
	"_name_or_path": "microsoft/cvt-21",
	"architectures": [
	"CvtForImageClassification"
	],
	"attention_drop_rate": [
	0.0,
	0.0,
	0.0
	],
	"cls_token": [
	false,
	false,
	true
	],
	"depth": [
	1,
	4,
	16
	],
	"drop_path_rate": [
	0.0,
	0.0,
	0.1
	],
	"drop_rate": [
	0.0,
	0.0,
	0.0
	],
	"embed_dim": [
	64,
	192,
	384
	],
	"id2label": {
	"0": "event",
	"1": "no_event"
	},
	"image_size": 224,
	"initializer_range": 0.02,
	"kernel_qkv": [
	3,
	3,
	3
	],
	"label2id": {
	"event": 0,
	"no_event": 1
	},
	"layer_norm_eps": 1e-12,
	"mlp_ratio": [
	4.0,
	4.0,
	4.0
	],
	"model_type": "cvt",
	"num_channels": 3,
	"num_heads": [
	1,
	3,
	6
	],
	"num_stages": 3,
	"padding_kv": [
	1,
	1,
	1
	],
	"padding_q": [
	1,
	1,
	1
	],
	"patch_padding": [
	2,
	1,
	1
	],
	"patch_sizes": [
	7,
	3,
	3
	],
	"patch_stride": [
	4,
	2,
	2
	],
	"pos_embed": [
	false,
	false,
	false
	],
	"problem_type": "single_label_classification",
	"qkv_bias": [
	true,
	true,
	true
	],
	"qkv_projection_method": [
	"dw_bn",
	"dw_bn",
	"dw_bn"
	],
	"stride_kv": [
	2,
	2,
	2
	],
	"stride_q": [
	1,
	1,
	1
	],
	"torch_dtype": "float32",
	"transformers_version": "4.24.0"
	}