swiftformer-xs / config.json
HorcruxNo13's picture
Training in progress, epoch 1
3a85ef9
raw
history blame
776 Bytes
{
"_name_or_path": "MBZUAI/swiftformer-xs",
"architectures": [
"SwiftFormerForImageClassification"
],
"batch_norm_eps": 1e-05,
"depths": [
3,
3,
6,
4
],
"down_pad": 1,
"down_patch_size": 3,
"down_stride": 2,
"downsamples": [
true,
true,
true,
true
],
"drop_path_rate": 0.0,
"embed_dims": [
48,
56,
112,
220
],
"hidden_act": "gelu",
"id2label": {
"0": "Normal",
"1": "Abnormal"
},
"label2id": {
"Abnormal": 1,
"Normal": 0
},
"layer_scale_init_value": 1e-05,
"mlp_ratio": 4,
"model_type": "swiftformer",
"num_channels": 3,
"problem_type": "single_label_classification",
"torch_dtype": "float32",
"transformers_version": "4.33.2",
"use_layer_scale": true
}