Augusto777's picture
Training in progress, epoch 0
6196075 verified
raw
history blame
900 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": "avanzada",
"1": "leve",
"2": "moderada",
"3": "no dmae"
},
"label2id": {
"avanzada": 0,
"leve": 1,
"moderada": 2,
"no dmae": 3
},
"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.36.2",
"use_layer_scale": true
}