From fcf529ef3252a204f61a3f54600eafd33df7f08a Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Tue, 23 Aug 2022 11:50:45 +0200 Subject: [PATCH] Adding a simple model. --- config.json | 93 +++++++++++++++++++++++++++++++++++++++++++++++ pytorch_model.bin | 3 ++ 2 files changed, 96 insertions(+) create mode 100644 config.json create mode 100644 pytorch_model.bin diff --git a/config.json b/config.json new file mode 100644 index 0000000..4520898 --- /dev/null +++ b/config.json @@ -0,0 +1,93 @@ +{ + "architectures": [ + "LayoutLMv2ForQuestionAnswering" + ], + "attention_probs_dropout_prob": 0.1, + "convert_sync_batchnorm": true, + "coordinate_size": 16, + "detectron2_config_args": { + "MODEL.ANCHOR_GENERATOR.SIZES": [ + [ + 32 + ] + ], + "MODEL.BACKBONE.NAME": "build_resnet_fpn_backbone", + "MODEL.FPN.IN_FEATURES": [ + "res2" + ], + "MODEL.MASK_ON": true, + "MODEL.PIXEL_STD": [ + 57.375, + 57.12, + 58.395 + ], + "MODEL.POST_NMS_TOPK_TEST": 1000, + "MODEL.RESNETS.ASPECT_RATIOS": [ + [ + 0.5, + 1.0, + 2.0 + ] + ], + "MODEL.RESNETS.DEPTH": 101, + "MODEL.RESNETS.NUM_GROUPS": 32, + "MODEL.RESNETS.OUT_FEATURES": [ + "res2" + ], + "MODEL.RESNETS.SIZES": [ + [ + 32 + ] + ], + "MODEL.RESNETS.STRIDE_IN_1X1": false, + "MODEL.RESNETS.WIDTH_PER_GROUP": 8, + "MODEL.ROI_BOX_HEAD.NAME": "FastRCNNConvFCHead", + "MODEL.ROI_BOX_HEAD.NUM_FC": 2, + "MODEL.ROI_BOX_HEAD.POOLER_RESOLUTION": 14, + "MODEL.ROI_HEADS.IN_FEATURES": [ + "p2" + ], + "MODEL.ROI_HEADS.NAME": "StandardROIHeads", + "MODEL.ROI_HEADS.NUM_CLASSES": 5, + "MODEL.ROI_MASK_HEAD.NAME": "MaskRCNNConvUpsampleHead", + "MODEL.ROI_MASK_HEAD.NUM_CONV": 4, + "MODEL.ROI_MASK_HEAD.POOLER_RESOLUTION": 7, + "MODEL.RPN.IN_FEATURES": [ + "p2", + "p3" + ], + "MODEL.RPN.POST_NMS_TOPK_TRAIN": 1000, + "MODEL.RPN.PRE_NMS_TOPK_TEST": 1000, + "MODEL.RPN.PRE_NMS_TOPK_TRAIN": 2000 + }, + "fast_qkv": true, + "has_relative_attention_bias": true, + "has_spatial_attention_bias": true, + "has_visual_segment_embedding": true, + "hidden_act": "gelu", + "hidden_dropout_prob": 0.1, + "hidden_size": 96, + "image_feature_pool_shape": [ + 2, + 2, + 256 + ], + "initializer_range": 0.02, + "intermediate_size": 64, + "layer_norm_eps": 1e-12, + "max_2d_position_embeddings": 1024, + "max_position_embeddings": 512, + "max_rel_2d_pos": 256, + "max_rel_pos": 128, + "model_type": "layoutlmv2", + "num_attention_heads": 4, + "num_hidden_layers": 2, + "pad_token_id": 0, + "rel_2d_pos_bins": 64, + "rel_pos_bins": 32, + "shape_size": 16, + "torch_dtype": "float32", + "transformers_version": "4.22.0.dev0", + "type_vocab_size": 2, + "vocab_size": 30522 +} diff --git a/pytorch_model.bin b/pytorch_model.bin new file mode 100644 index 0000000..4e5ea53 --- /dev/null +++ b/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:78b808255d4d8ee9c3c7350bcc35ed50aa9970b0225308b1c98da1a02caba869 +size 17126119