From c0b18810c5af73193f9830896e0f8be8cc93d080 Mon Sep 17 00:00:00 2001 From: Niels Rogge Date: Fri, 18 Mar 2022 16:15:26 +0100 Subject: [PATCH] Add model --- config.json | 4 ++-- pytorch_model.bin | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/config.json b/config.json index d119fe3..b45e3db 100644 --- a/config.json +++ b/config.json @@ -4,7 +4,7 @@ ], "attention_probs_dropout_prob": 0.0, "classifier_dropout_prob": 0.1, - "decoder_hidden_size": 768, + "decoder_hidden_size": 64, "depths": [ 3, 8, @@ -18,6 +18,7 @@ 16 ], "drop_path_rate": 0.1, + "head_in_index": -1, "hidden_act": "gelu", "hidden_dropout_prob": 0.0, "hidden_sizes": [ @@ -27,7 +28,6 @@ 512 ], "image_size": 224, - "in_index": -1, "initializer_range": 0.02, "layer_norm_eps": 1e-06, "max_depth": 10, diff --git a/pytorch_model.bin b/pytorch_model.bin index 213fbad..80ef813 100644 --- a/pytorch_model.bin +++ b/pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:7785ad1a9c126fb0eec67142b4d226c8e8f537b884345cbf046138238026fc39 -size 245256489 +oid sha256:dbb5f777e542329280f1decc1bd090f8f770e770dca6fc4407c865abcef7e84f +size 245258793