From 72d580cae9d3a854630da97a8cfd8e42ab97dbca Mon Sep 17 00:00:00 2001 From: Niels Rogge Date: Fri, 2 Dec 2022 08:13:34 +0000 Subject: [PATCH] update weight and config to reflect latest changes from the pr (#1) - update weight and config to reflect latest changes from the pr (4be000e6aef4a3488666ca88c2ab1a6714c69c07) Co-authored-by: Fatih --- config.json | 7 +++---- preprocessor_config.json | 26 ++++++++++++++++++++++++++ pytorch_model.bin | 4 ++-- 3 files changed, 31 insertions(+), 6 deletions(-) create mode 100644 preprocessor_config.json diff --git a/config.json b/config.json index 2c0ba01..71d4870 100644 --- a/config.json +++ b/config.json @@ -1,10 +1,10 @@ { "architectures": [ - "TimeSformerForVideoClassification" + "TimesformerForVideoClassification" ], "attention_probs_dropout_prob": 0.0, "attention_type": "divided_space_time", - "drop_path_prob": 0, + "drop_path_rate": 0, "hidden_act": "gelu", "hidden_dropout_prob": 0.0, "hidden_size": 768, @@ -824,6 +824,5 @@ "patch_size": 16, "qkv_bias": true, "torch_dtype": "float32", - "transformers_version": "4.23.0.dev0", - "use_mean_pooling": false + "transformers_version": "4.25.0.dev0" } diff --git a/preprocessor_config.json b/preprocessor_config.json new file mode 100644 index 0000000..05f8840 --- /dev/null +++ b/preprocessor_config.json @@ -0,0 +1,26 @@ +{ + "crop_size": { + "height": 224, + "width": 224 + }, + "do_center_crop": true, + "do_normalize": true, + "do_rescale": true, + "do_resize": true, + "image_mean": [ + 0.5, + 0.5, + 0.5 + ], + "image_processor_type": "VideoMAEImageProcessor", + "image_std": [ + 0.5, + 0.5, + 0.5 + ], + "resample": 2, + "rescale_factor": 0.00392156862745098, + "size": { + "shortest_edge": 224 + } +} diff --git a/pytorch_model.bin b/pytorch_model.bin index 329ec63..4110539 100644 --- a/pytorch_model.bin +++ b/pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:968479f41eb3f2f4b1829ae06b663998dcc3782deb6a3e81746a9e51dbabd46d -size 486346639 +oid sha256:bdb264469ffc423fb86c78932268a290bfbd69bd768b7ce8595c047a2dc54089 +size 486348721