diff --git a/config.json b/config.json new file mode 100644 index 0000000..2493202 --- /dev/null +++ b/config.json @@ -0,0 +1,82 @@ +{ + "architectures": [ + "BeitForImageClassification" + ], + "attention_probs_dropout_prob": 0.0, + "auxiliary_channels": 256, + "auxiliary_concat_input": false, + "auxiliary_loss_weight": 0.4, + "auxiliary_num_convs": 1, + "drop_path_rate": 0.1, + "hidden_act": "gelu", + "hidden_dropout_prob": 0.0, + "hidden_size": 1024, + "id2label": { + "0": "letter", + "1": "form", + "2": "email", + "3": "handwritten", + "4": "advertisement", + "5": "scientific report", + "6": "scientific publication", + "7": "specification", + "8": "file folder", + "9": "news article", + "10": "budget", + "11": "invoice", + "12": "presentation", + "13": "questionnaire", + "14": "resume", + "15": "memo" + }, + "image_size": 224, + "initializer_range": 0.02, + "intermediate_size": 4096, + "label2id": { + "advertisement": 4, + "budget": 10, + "email": 2, + "file folder": 8, + "form": 1, + "handwritten": 3, + "invoice": 11, + "letter": 0, + "memo": 15, + "news article": 9, + "presentation": 12, + "questionnaire": 13, + "resume": 14, + "scientific publication": 6, + "scientific report": 5, + "specification": 7 + }, + "layer_norm_eps": 1e-12, + "layer_scale_init_value": 0.1, + "model_type": "beit", + "num_attention_heads": 16, + "num_channels": 3, + "num_hidden_layers": 24, + "out_indices": [ + 3, + 5, + 7, + 11 + ], + "patch_size": 16, + "pool_scales": [ + 1, + 2, + 3, + 6 + ], + "semantic_loss_ignore_index": 255, + "torch_dtype": "float32", + "transformers_version": "4.18.0.dev0", + "use_absolute_position_embeddings": true, + "use_auxiliary_head": true, + "use_mask_token": false, + "use_mean_pooling": true, + "use_relative_position_bias": false, + "use_shared_relative_position_bias": false, + "vocab_size": 8192 +} diff --git a/preprocessor_config.json b/preprocessor_config.json new file mode 100644 index 0000000..8a3d975 --- /dev/null +++ b/preprocessor_config.json @@ -0,0 +1,20 @@ +{ + "crop_size": 224, + "do_center_crop": false, + "do_normalize": true, + "do_resize": true, + "feature_extractor_type": "BeitFeatureExtractor", + "image_mean": [ + 0.5, + 0.5, + 0.5 + ], + "image_std": [ + 0.5, + 0.5, + 0.5 + ], + "reduce_labels": false, + "resample": 2, + "size": 224 +} diff --git a/pytorch_model.bin b/pytorch_model.bin new file mode 100644 index 0000000..efe3385 --- /dev/null +++ b/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:819c39abe94292bfb9e2992a70053e49968b71563a352f156167b5f024499fac +size 1213532225