test / config.json
Gilsson's picture
Training in progress, epoch 1
ef8306e verified
raw
history blame
3.59 kB
{
"_name_or_path": "microsoft/conditional-detr-resnet-50",
"activation_dropout": 0.0,
"activation_function": "relu",
"architectures": [
"ConditionalDetrForObjectDetection"
],
"attention_dropout": 0.0,
"auxiliary_loss": false,
"backbone": "resnet50",
"backbone_config": null,
"backbone_kwargs": {
"in_chans": 3,
"out_indices": [
1,
2,
3,
4
]
},
"bbox_cost": 5,
"bbox_loss_coefficient": 5,
"class_cost": 2,
"cls_loss_coefficient": 2,
"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,
"focal_alpha": 0.25,
"giou_cost": 2,
"giou_loss_coefficient": 2,
"id2label": {
"0": "acrid",
"1": "ancient wisp",
"2": "archer bug",
"3": "arhaic wisp",
"4": "bison",
"5": "boarlit",
"6": "bramble",
"7": "child",
"8": "clay man",
"9": "colossus",
"10": "crab",
"11": "cremator",
"12": "direseeker",
"13": "elder lemurian",
"14": "evolved lemurian",
"15": "gilded wurms",
"16": "golem",
"17": "greater wisp",
"18": "gup",
"19": "ifrit",
"20": "imp",
"21": "imp overlord",
"22": "jellyfish",
"23": "lemurian",
"24": "lemurian cavalry",
"25": "lunx tribe",
"26": "lynx totem",
"27": "macrobe",
"28": "macrobic predator",
"29": "magma worm",
"30": "mushrum",
"31": "parent",
"32": "providence",
"33": "purple imp",
"34": "sanctuary guard",
"35": "scavenger",
"36": "spider",
"37": "spitter",
"38": "swift",
"39": "temple guard",
"40": "tiny imp",
"41": "toxic beast",
"42": "trokk",
"43": "tuber",
"44": "wandering vagrant",
"45": "whorl",
"46": "wisp",
"47": "young vagrant"
},
"init_std": 0.02,
"init_xavier_std": 1.0,
"is_encoder_decoder": true,
"label2id": {
"acrid": 0,
"ancient wisp": 1,
"archer bug": 2,
"arhaic wisp": 3,
"bison": 4,
"boarlit": 5,
"bramble": 6,
"child": 7,
"clay man": 8,
"colossus": 9,
"crab": 10,
"cremator": 11,
"direseeker": 12,
"elder lemurian": 13,
"evolved lemurian": 14,
"gilded wurms": 15,
"golem": 16,
"greater wisp": 17,
"gup": 18,
"ifrit": 19,
"imp": 20,
"imp overlord": 21,
"jellyfish": 22,
"lemurian": 23,
"lemurian cavalry": 24,
"lunx tribe": 25,
"lynx totem": 26,
"macrobe": 27,
"macrobic predator": 28,
"magma worm": 29,
"mushrum": 30,
"parent": 31,
"providence": 32,
"purple imp": 33,
"sanctuary guard": 34,
"scavenger": 35,
"spider": 36,
"spitter": 37,
"swift": 38,
"temple guard": 39,
"tiny imp": 40,
"toxic beast": 41,
"trokk": 42,
"tuber": 43,
"wandering vagrant": 44,
"whorl": 45,
"wisp": 46,
"young vagrant": 47
},
"mask_loss_coefficient": 1,
"max_position_embeddings": 1024,
"model_type": "conditional_detr",
"num_channels": 3,
"num_hidden_layers": 6,
"num_queries": 300,
"position_embedding_type": "sine",
"scale_embedding": false,
"torch_dtype": "float32",
"transformers_version": "4.46.1",
"use_pretrained_backbone": true,
"use_timm_backbone": true
}