amaye15's picture
End of training
84d24c2 verified
{
"_name_or_path": "microsoft/swinv2-base-patch4-window16-256",
"architectures": [
"Swinv2ForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"depths": [
2,
2,
18,
2
],
"drop_path_rate": 0.1,
"embed_dim": 128,
"encoder_stride": 32,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 1024,
"id2label": {
"0": "Affenpinscher",
"1": "Afghan Hound",
"2": "African Hunting Dog",
"3": "Airedale",
"4": "American Staffordshire Terrier",
"5": "Appenzeller",
"6": "Australian Terrier",
"7": "Basenji",
"8": "Basset",
"9": "Beagle",
"10": "Bedlington Terrier",
"11": "Bernese Mountain Dog",
"12": "Black And Tan Coonhound",
"13": "Blenheim Spaniel",
"14": "Bloodhound",
"15": "Bluetick",
"16": "Border Collie",
"17": "Border Terrier",
"18": "Borzoi",
"19": "Boston Bull",
"20": "Bouvier Des Flandres",
"21": "Boxer",
"22": "Brabancon Griffon",
"23": "Briard",
"24": "Brittany Spaniel",
"25": "Bull Mastiff",
"26": "Cairn",
"27": "Cardigan",
"28": "Chesapeake Bay Retriever",
"29": "Chihuahua",
"30": "Chow",
"31": "Clumber",
"32": "Cocker Spaniel",
"33": "Collie",
"34": "Curly Coated Retriever",
"35": "Dandie Dinmont",
"36": "Dhole",
"37": "Dingo",
"38": "Doberman",
"39": "English Foxhound",
"40": "English Setter",
"41": "English Springer",
"42": "Entlebucher",
"43": "Eskimo Dog",
"44": "Flat Coated Retriever",
"45": "French Bulldog",
"46": "German Shepherd",
"47": "German Short Haired Pointer",
"48": "Giant Schnauzer",
"49": "Golden Retriever",
"50": "Gordon Setter",
"51": "Great Dane",
"52": "Great Pyrenees",
"53": "Greater Swiss Mountain Dog",
"54": "Groenendael",
"55": "Ibizan Hound",
"56": "Irish Setter",
"57": "Irish Terrier",
"58": "Irish Water Spaniel",
"59": "Irish Wolfhound",
"60": "Italian Greyhound",
"61": "Japanese Spaniel",
"62": "Keeshond",
"63": "Kelpie",
"64": "Kerry Blue Terrier",
"65": "Komondor",
"66": "Kuvasz",
"67": "Labrador Retriever",
"68": "Lakeland Terrier",
"69": "Leonberg",
"70": "Lhasa",
"71": "Malamute",
"72": "Malinois",
"73": "Maltese Dog",
"74": "Mexican Hairless",
"75": "Miniature Pinscher",
"76": "Miniature Poodle",
"77": "Miniature Schnauzer",
"78": "Newfoundland",
"79": "Norfolk Terrier",
"80": "Norwegian Elkhound",
"81": "Norwich Terrier",
"82": "Old English Sheepdog",
"83": "Otterhound",
"84": "Papillon",
"85": "Pekinese",
"86": "Pembroke",
"87": "Pomeranian",
"88": "Pug",
"89": "Redbone",
"90": "Rhodesian Ridgeback",
"91": "Rottweiler",
"92": "Saint Bernard",
"93": "Saluki",
"94": "Samoyed",
"95": "Schipperke",
"96": "Scotch Terrier",
"97": "Scottish Deerhound",
"98": "Sealyham Terrier",
"99": "Shetland Sheepdog",
"100": "Shih Tzu",
"101": "Siberian Husky",
"102": "Silky Terrier",
"103": "Soft Coated Wheaten Terrier",
"104": "Staffordshire Bullterrier",
"105": "Standard Poodle",
"106": "Standard Schnauzer",
"107": "Sussex Spaniel",
"108": "Tibetan Mastiff",
"109": "Tibetan Terrier",
"110": "Toy Poodle",
"111": "Toy Terrier",
"112": "Vizsla",
"113": "Walker Hound",
"114": "Weimaraner",
"115": "Welsh Springer Spaniel",
"116": "West Highland White Terrier",
"117": "Whippet",
"118": "Wire Haired Fox Terrier",
"119": "Yorkshire Terrier"
},
"image_size": 256,
"initializer_range": 0.02,
"label2id": {
"Affenpinscher": 0,
"Afghan Hound": 1,
"African Hunting Dog": 2,
"Airedale": 3,
"American Staffordshire Terrier": 4,
"Appenzeller": 5,
"Australian Terrier": 6,
"Basenji": 7,
"Basset": 8,
"Beagle": 9,
"Bedlington Terrier": 10,
"Bernese Mountain Dog": 11,
"Black And Tan Coonhound": 12,
"Blenheim Spaniel": 13,
"Bloodhound": 14,
"Bluetick": 15,
"Border Collie": 16,
"Border Terrier": 17,
"Borzoi": 18,
"Boston Bull": 19,
"Bouvier Des Flandres": 20,
"Boxer": 21,
"Brabancon Griffon": 22,
"Briard": 23,
"Brittany Spaniel": 24,
"Bull Mastiff": 25,
"Cairn": 26,
"Cardigan": 27,
"Chesapeake Bay Retriever": 28,
"Chihuahua": 29,
"Chow": 30,
"Clumber": 31,
"Cocker Spaniel": 32,
"Collie": 33,
"Curly Coated Retriever": 34,
"Dandie Dinmont": 35,
"Dhole": 36,
"Dingo": 37,
"Doberman": 38,
"English Foxhound": 39,
"English Setter": 40,
"English Springer": 41,
"Entlebucher": 42,
"Eskimo Dog": 43,
"Flat Coated Retriever": 44,
"French Bulldog": 45,
"German Shepherd": 46,
"German Short Haired Pointer": 47,
"Giant Schnauzer": 48,
"Golden Retriever": 49,
"Gordon Setter": 50,
"Great Dane": 51,
"Great Pyrenees": 52,
"Greater Swiss Mountain Dog": 53,
"Groenendael": 54,
"Ibizan Hound": 55,
"Irish Setter": 56,
"Irish Terrier": 57,
"Irish Water Spaniel": 58,
"Irish Wolfhound": 59,
"Italian Greyhound": 60,
"Japanese Spaniel": 61,
"Keeshond": 62,
"Kelpie": 63,
"Kerry Blue Terrier": 64,
"Komondor": 65,
"Kuvasz": 66,
"Labrador Retriever": 67,
"Lakeland Terrier": 68,
"Leonberg": 69,
"Lhasa": 70,
"Malamute": 71,
"Malinois": 72,
"Maltese Dog": 73,
"Mexican Hairless": 74,
"Miniature Pinscher": 75,
"Miniature Poodle": 76,
"Miniature Schnauzer": 77,
"Newfoundland": 78,
"Norfolk Terrier": 79,
"Norwegian Elkhound": 80,
"Norwich Terrier": 81,
"Old English Sheepdog": 82,
"Otterhound": 83,
"Papillon": 84,
"Pekinese": 85,
"Pembroke": 86,
"Pomeranian": 87,
"Pug": 88,
"Redbone": 89,
"Rhodesian Ridgeback": 90,
"Rottweiler": 91,
"Saint Bernard": 92,
"Saluki": 93,
"Samoyed": 94,
"Schipperke": 95,
"Scotch Terrier": 96,
"Scottish Deerhound": 97,
"Sealyham Terrier": 98,
"Shetland Sheepdog": 99,
"Shih Tzu": 100,
"Siberian Husky": 101,
"Silky Terrier": 102,
"Soft Coated Wheaten Terrier": 103,
"Staffordshire Bullterrier": 104,
"Standard Poodle": 105,
"Standard Schnauzer": 106,
"Sussex Spaniel": 107,
"Tibetan Mastiff": 108,
"Tibetan Terrier": 109,
"Toy Poodle": 110,
"Toy Terrier": 111,
"Vizsla": 112,
"Walker Hound": 113,
"Weimaraner": 114,
"Welsh Springer Spaniel": 115,
"West Highland White Terrier": 116,
"Whippet": 117,
"Wire Haired Fox Terrier": 118,
"Yorkshire Terrier": 119
},
"layer_norm_eps": 1e-05,
"mlp_ratio": 4.0,
"model_type": "swinv2",
"num_channels": 3,
"num_heads": [
4,
8,
16,
32
],
"num_layers": 4,
"out_features": [
"stage4"
],
"out_indices": [
4
],
"patch_size": 4,
"path_norm": true,
"pretrained_window_sizes": [
0,
0,
0,
0
],
"problem_type": "single_label_classification",
"qkv_bias": true,
"stage_names": [
"stem",
"stage1",
"stage2",
"stage3",
"stage4"
],
"torch_dtype": "float32",
"transformers_version": "4.40.2",
"use_absolute_embeddings": false,
"window_size": 16
}