hf-transformers-bot's picture
Update tiny models for TableTransformerModel
147a953 verified
Raw
History Blame Contribute Delete
1.88 kB
{
"activation_dropout": 0.0,
"activation_function": "relu",
"architectures": [
"TableTransformerModel"
],
"attention_dropout": 0.1,
"auxiliary_loss": false,
"backbone": null,
"backbone_config": {
"depths": [
1,
1,
2,
1
],
"downsample_in_bottleneck": false,
"downsample_in_first_stage": false,
"embedding_size": 64,
"embeddings_size": 10,
"hidden_act": "relu",
"hidden_sizes": [
10,
20,
30,
40
],
"id2label": {
"0": "LABEL_0",
"1": "LABEL_1",
"2": "LABEL_2"
},
"label2id": {
"LABEL_0": 0,
"LABEL_1": 1,
"LABEL_2": 2
},
"layer_type": "bottleneck",
"model_type": "resnet",
"num_channels": 3,
"out_features": [
"stage2",
"stage3",
"stage4"
],
"out_indices": [
2,
3,
4
],
"stage_names": [
"stem",
"stage1",
"stage2",
"stage3",
"stage4"
]
},
"bbox_cost": 5,
"bbox_loss_coefficient": 5,
"class_cost": 1,
"d_model": 32,
"decoder_attention_heads": 8,
"decoder_ffn_dim": 4,
"decoder_layerdrop": 0.0,
"decoder_layers": 2,
"dice_loss_coefficient": 1,
"dilation": false,
"dropout": 0.1,
"dtype": "float32",
"encoder_attention_heads": 8,
"encoder_ffn_dim": 4,
"encoder_layerdrop": 0.0,
"encoder_layers": 2,
"eos_coefficient": 0.1,
"giou_cost": 2,
"giou_loss_coefficient": 2,
"id2label": {
"0": "LABEL_0",
"1": "LABEL_1",
"2": "LABEL_2"
},
"init_std": 0.02,
"init_xavier_std": 1.0,
"is_encoder_decoder": true,
"label2id": {
"LABEL_0": 0,
"LABEL_1": 1,
"LABEL_2": 2
},
"mask_loss_coefficient": 1,
"model_type": "table-transformer",
"num_channels": 3,
"num_queries": 12,
"position_embedding_type": "sine",
"transformers_version": "5.16.0.dev0"
}