Team Ai
Modelpublic

KakashiH/Image_Based_Malware_Classification_ViT

sourceHugging Faceapache-2.0updated 2y agoView on Hugging Face
1likes
config.json47 linesDownload Raw Back to root
1{2  "_name_or_path": "google/vit-base-patch16-224",3  "architectures": [4    "ViTForImageClassification"5  ],6  "attention_probs_dropout_prob": 0.0,7  "encoder_stride": 16,8  "hidden_act": "gelu",9  "hidden_dropout_prob": 0.0,10  "hidden_size": 768,11  "id2label": {12    "0": "Gatak",13    "1": "Kelihos_ver1",14    "2": "Kelihos_ver3",15    "3": "Lollipop",16    "4": "Obfuscator.ACY",17    "5": "Ramnit",18    "6": "Simda",19    "7": "Tracur",20    "8": "Vundo"21  },22  "image_size": 224,23  "initializer_range": 0.02,24  "intermediate_size": 3072,25  "label2id": {26    "Gatak": 0,27    "Kelihos_ver1": 1,28    "Kelihos_ver3": 2,29    "Lollipop": 3,30    "Obfuscator.ACY": 4,31    "Ramnit": 5,32    "Simda": 6,33    "Tracur": 7,34    "Vundo": 835  },36  "layer_norm_eps": 1e-12,37  "model_type": "vit",38  "num_attention_heads": 12,39  "num_channels": 3,40  "num_hidden_layers": 12,41  "patch_size": 16,42  "problem_type": "single_label_classification",43  "qkv_bias": true,44  "torch_dtype": "float32",45  "transformers_version": "4.38.2"46}47