From 9f7d1696ea4adc5cdc5c7cb1ec0397b66231cf21 Mon Sep 17 00:00:00 2001 From: Niels Rogge Date: Tue, 22 Mar 2022 14:29:40 +0100 Subject: [PATCH] Add model --- config.json | 18 +++++++++--------- pytorch_model.bin | 4 ++-- 2 files changed, 11 insertions(+), 11 deletions(-) diff --git a/config.json b/config.json index 689be59..ee679a1 100644 --- a/config.json +++ b/config.json @@ -4,12 +4,18 @@ ], "attention_probs_dropout_prob": 0.0, "auxiliary_loss_weight": 0.4, - "channels": 256, + "backbone_out_indices": [ + 5, + 11, + 17, + 23 + ], + "fusion_hidden_size": 256, + "head_in_index": -1, "hidden_act": "gelu", "hidden_dropout_prob": 0.0, "hidden_size": 1024, "image_size": 384, - "in_index": -1, "initializer_range": 0.02, "intermediate_size": 4096, "layer_norm_eps": 1e-12, @@ -23,12 +29,6 @@ "num_attention_heads": 16, "num_channels": 3, "num_hidden_layers": 24, - "out_indices": [ - 5, - 11, - 17, - 23 - ], "patch_size": 16, "qkv_bias": true, "readout_type": "project", @@ -43,5 +43,5 @@ "torch_dtype": "float32", "transformers_version": "4.18.0.dev0", "use_auxiliary_head": true, - "use_batch_norm": false + "use_batch_norm_in_fusion_residual": false } diff --git a/pytorch_model.bin b/pytorch_model.bin index f061b09..53ee22f 100644 --- a/pytorch_model.bin +++ b/pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:434f2e147724ab260492e3f4d33ae6629da181957f05f2d503e7f4a4e0bf34f0 -size 1367580589 +oid sha256:71150941604c39c1c770a72a7b2f56487669f0e3a4d99c8759e233fb1be24080 +size 1367581165