| { | |
| "_name_or_path": "distilbert-base-uncased", | |
| "activation": "gelu", | |
| "architectures": [ | |
| "DistilBertForSequenceClassification" | |
| ], | |
| "attention_dropout": 0.1, | |
| "dim": 768, | |
| "dropout": 0.1, | |
| "hidden_dim": 3072, | |
| "id2label": { | |
| "0": "Headwear", | |
| "1": "Topwear", | |
| "2": "Accessories", | |
| "3": "Cufflinks", | |
| "4": "Free Gifts", | |
| "5": "Apparel Set", | |
| "6": "Watches", | |
| "7": "Bottomwear", | |
| "8": "Eyewear", | |
| "9": "Lips", | |
| "10": "Innerwear", | |
| "11": "Ties", | |
| "12": "Wallets", | |
| "13": "Nails", | |
| "14": "Loungewear and Nightwear", | |
| "15": "Dress", | |
| "16": "Fragrance", | |
| "17": "Bags", | |
| "18": "Socks", | |
| "19": "Jewellery", | |
| "20": "Makeup", | |
| "21": "Scarves", | |
| "22": "Flip Flops", | |
| "23": "Sandal", | |
| "24": "Saree", | |
| "25": "Belts", | |
| "26": "Shoes" | |
| }, | |
| "initializer_range": 0.02, | |
| "label2id": { | |
| "Accessories": 2, | |
| "Apparel Set": 5, | |
| "Bags": 17, | |
| "Belts": 25, | |
| "Bottomwear": 7, | |
| "Cufflinks": 3, | |
| "Dress": 15, | |
| "Eyewear": 8, | |
| "Flip Flops": 22, | |
| "Fragrance": 16, | |
| "Free Gifts": 4, | |
| "Headwear": 0, | |
| "Innerwear": 10, | |
| "Jewellery": 19, | |
| "Lips": 9, | |
| "Loungewear and Nightwear": 14, | |
| "Makeup": 20, | |
| "Nails": 13, | |
| "Sandal": 23, | |
| "Saree": 24, | |
| "Scarves": 21, | |
| "Shoes": 26, | |
| "Socks": 18, | |
| "Ties": 11, | |
| "Topwear": 1, | |
| "Wallets": 12, | |
| "Watches": 6 | |
| }, | |
| "max_position_embeddings": 512, | |
| "model_type": "distilbert", | |
| "n_heads": 12, | |
| "n_layers": 6, | |
| "pad_token_id": 0, | |
| "problem_type": "single_label_classification", | |
| "qa_dropout": 0.1, | |
| "seq_classif_dropout": 0.2, | |
| "sinusoidal_pos_embds": false, | |
| "tie_weights_": true, | |
| "torch_dtype": "float32", | |
| "transformers_version": "4.28.0", | |
| "vocab_size": 30522 | |
| } | |