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