train_5ep_test / config.json
whyoke's picture
Training in progress, step 200
c9236d7
{
"_name_or_path": "facebook/detr-resnet-50",
"activation_dropout": 0.0,
"activation_function": "relu",
"architectures": [
"DetrForObjectDetection"
],
"attention_dropout": 0.0,
"auxiliary_loss": false,
"backbone": "resnet50",
"backbone_config": null,
"bbox_cost": 5,
"bbox_loss_coefficient": 5,
"class_cost": 1,
"d_model": 256,
"decoder_attention_heads": 8,
"decoder_ffn_dim": 2048,
"decoder_layerdrop": 0.0,
"decoder_layers": 6,
"dice_loss_coefficient": 1,
"dilation": false,
"dropout": 0.1,
"encoder_attention_heads": 8,
"encoder_ffn_dim": 2048,
"encoder_layerdrop": 0.0,
"encoder_layers": 6,
"eos_coefficient": 0.1,
"giou_cost": 2,
"giou_loss_coefficient": 2,
"id2label": {
"0": "Acne_Cosmetica",
"1": "Acne_Erytherma",
"2": "Acne_Milia",
"3": "Acne_PIH",
"4": "Acne_T-spinulusa",
"5": "Acne_acne",
"6": "Acne_estivalis",
"7": "Acne_nodule",
"8": "Acne_pitted",
"9": "Calcosis Cutis",
"10": "Dilate pore",
"11": "Freckles_Deep",
"12": "Freckles_Sebker",
"13": "Freckles_Solar",
"14": "Freckles_freckles",
"15": "Hemangioma",
"16": "Keratosis",
"17": "Lentigene",
"18": "Melasma",
"19": "Melasma_likeTelangiectasia",
"20": "Nevus",
"21": "PIH_lip",
"22": "Seb_Gland",
"23": "Syringoma",
"24": "Telangiectasia",
"25": "allergic_contact_dermatitis",
"26": "atopy",
"27": "cyst",
"28": "herpes_zoster",
"29": "hifu_tightening",
"30": "hypertrophic_scar",
"31": "irritant_contact_dermatitis",
"32": "keloid",
"33": "painting",
"34": "perioral_dermatitis",
"35": "pitted_scar",
"36": "portwine_stain",
"37": "pricken_nodale",
"38": "psoriasis",
"39": "rejuvenation",
"40": "resurface",
"41": "sebderm",
"42": "segmental_lentigene",
"43": "steroid_addict",
"44": "stretch_mark",
"45": "tattoo",
"46": "tinea_unguium",
"47": "xanthelasma"
},
"init_std": 0.02,
"init_xavier_std": 1.0,
"is_encoder_decoder": true,
"label2id": {
"Acne_Cosmetica": 0,
"Acne_Erytherma": 1,
"Acne_Milia": 2,
"Acne_PIH": 3,
"Acne_T-spinulusa": 4,
"Acne_acne": 5,
"Acne_estivalis": 6,
"Acne_nodule": 7,
"Acne_pitted": 8,
"Calcosis Cutis": 9,
"Dilate pore": 10,
"Freckles_Deep": 11,
"Freckles_Sebker": 12,
"Freckles_Solar": 13,
"Freckles_freckles": 14,
"Hemangioma": 15,
"Keratosis": 16,
"Lentigene": 17,
"Melasma": 18,
"Melasma_likeTelangiectasia": 19,
"Nevus": 20,
"PIH_lip": 21,
"Seb_Gland": 22,
"Syringoma": 23,
"Telangiectasia": 24,
"allergic_contact_dermatitis": 25,
"atopy": 26,
"cyst": 27,
"herpes_zoster": 28,
"hifu_tightening": 29,
"hypertrophic_scar": 30,
"irritant_contact_dermatitis": 31,
"keloid": 32,
"painting": 33,
"perioral_dermatitis": 34,
"pitted_scar": 35,
"portwine_stain": 36,
"pricken_nodale": 37,
"psoriasis": 38,
"rejuvenation": 39,
"resurface": 40,
"sebderm": 41,
"segmental_lentigene": 42,
"steroid_addict": 43,
"stretch_mark": 44,
"tattoo": 45,
"tinea_unguium": 46,
"xanthelasma": 47
},
"mask_loss_coefficient": 1,
"max_position_embeddings": 1024,
"model_type": "detr",
"num_channels": 3,
"num_hidden_layers": 6,
"num_queries": 100,
"position_embedding_type": "sine",
"scale_embedding": false,
"torch_dtype": "float32",
"transformers_version": "4.26.1",
"use_pretrained_backbone": true,
"use_timm_backbone": true
}