config.json

{
  "_name_or_path": "microsoft/swinv2-large-patch4-window12to24-192to384-22kto1k-ft",
  "architectures": [
    "Swinv2ForImageClassification"
  ],
  "attention_probs_dropout_prob": 0.0,
  "depths": [
    2,
    2,
    18,
    2
  ],
  "drop_path_rate": 0.1,
  "embed_dim": 192,
  "encoder_stride": 32,
  "hidden_act": "gelu",
  "hidden_dropout_prob": 0.0,
  "hidden_size": 1536,
  "id2label": {
    "0": "Actinomycetes_Mycolata",
    "1": "Anaerobic Type B",
    "2": "Anaerobic suflur bacteria",
    "3": "Beggiatoa",
    "4": "Flexibacter",
    "5": "Fungi",
    "6": "GAO",
    "7": "Haliscomenobacter",
    "8": "Hyphomicrobium",
    "9": "Microscrilla",
    "10": "Microthrix",
    "11": "Nitrosomonas",
    "12": "Nostocoida limicola III",
    "13": "Nostocoida limicola I_II",
    "14": "PAO",
    "15": "Sphaerotilus",
    "16": "Spirilla",
    "17": "Spirochaetes",
    "18": "Tetrads",
    "19": "Thiothrix",
    "20": "Type 0041_0675",
    "21": "Type 0092",
    "22": "Type 021N",
    "23": "Type 0411",
    "24": "Type 0581_Chloroflexi",
    "25": "Type 0914_0803",
    "26": "Type 0961",
    "27": "Type 1701",
    "28": "Type 1863",
    "29": "Zoogloea",
    "30": "bristleworm",
    "31": "crustacean",
    "32": "dead filament",
    "33": "elevated polysaccharide",
    "34": "ferric iron",
    "35": "fibrous material",
    "36": "filament impact on floc structure high",
    "37": "filament impact on floc structure low",
    "38": "filament impact on floc structure moderate",
    "39": "flagellate",
    "40": "floc open_diffuse",
    "41": "floc strong",
    "42": "floc weak",
    "43": "free swimming ciliate",
    "44": "gastrotrich",
    "45": "grease",
    "46": "inert",
    "47": "iron sulfide",
    "48": "irregular growth formations",
    "49": "naked amoebae",
    "50": "nematode",
    "51": "normal polysaccharide",
    "52": "oil",
    "53": "rotifer",
    "54": "stalked ciliate",
    "55": "testate amoebae",
    "56": "type 1851",
    "57": "water bear",
    "58": "yeast"
  },
  "image_size": 384,
  "initializer_range": 0.02,
  "label2id": {
    "Actinomycetes_Mycolata": 0,
    "Anaerobic Type B": 1,
    "Anaerobic suflur bacteria": 2,
    "Beggiatoa": 3,
    "Flexibacter": 4,
    "Fungi": 5,
    "GAO": 6,
    "Haliscomenobacter": 7,
    "Hyphomicrobium": 8,
    "Microscrilla": 9,
    "Microthrix": 10,
    "Nitrosomonas": 11,
    "Nostocoida limicola III": 12,
    "Nostocoida limicola I_II": 13,
    "PAO": 14,
    "Sphaerotilus": 15,
    "Spirilla": 16,
    "Spirochaetes": 17,
    "Tetrads": 18,
    "Thiothrix": 19,
    "Type 0041_0675": 20,
    "Type 0092": 21,
    "Type 021N": 22,
    "Type 0411": 23,
    "Type 0581_Chloroflexi": 24,
    "Type 0914_0803": 25,
    "Type 0961": 26,
    "Type 1701": 27,
    "Type 1863": 28,
    "Zoogloea": 29,
    "bristleworm": 30,
    "crustacean": 31,
    "dead filament": 32,
    "elevated polysaccharide": 33,
    "ferric iron": 34,
    "fibrous material": 35,
    "filament impact on floc structure high": 36,
    "filament impact on floc structure low": 37,
    "filament impact on floc structure moderate": 38,
    "flagellate": 39,
    "floc open_diffuse": 40,
    "floc strong": 41,
    "floc weak": 42,
    "free swimming ciliate": 43,
    "gastrotrich": 44,
    "grease": 45,
    "inert": 46,
    "iron sulfide": 47,
    "irregular growth formations": 48,
    "naked amoebae": 49,
    "nematode": 50,
    "normal polysaccharide": 51,
    "oil": 52,
    "rotifer": 53,
    "stalked ciliate": 54,
    "testate amoebae": 55,
    "type 1851": 56,
    "water bear": 57,
    "yeast": 58
  },
  "layer_norm_eps": 1e-05,
  "mlp_ratio": 4.0,
  "model_type": "swinv2",
  "num_channels": 3,
  "num_heads": [
    6,
    12,
    24,
    48
  ],
  "num_layers": 4,
  "patch_size": 4,
  "path_norm": true,
  "pretrained_window_sizes": [
    0,
    0,
    0,
    0
  ],
  "problem_type": "single_label_classification",
  "qkv_bias": true,
  "torch_dtype": "float32",
  "transformers_version": "4.33.1",
  "use_absolute_embeddings": false,
  "window_size": 24
}