From 580fedb8f5a62ab218620847e9177425be2fe366 Mon Sep 17 00:00:00 2001 From: yang yingjie Date: Mon, 17 Feb 2025 16:38:05 +0800 Subject: [PATCH] first commit --- config.json | 29 +++++++++++++++++++++++++++++ model.safetensors | 3 +++ preprocessor_config.json | 18 ++++++++++++++++++ pytorch_model.bin | 3 +++ 4 files changed, 53 insertions(+) create mode 100644 config.json create mode 100644 model.safetensors create mode 100644 preprocessor_config.json create mode 100644 pytorch_model.bin diff --git a/config.json b/config.json new file mode 100644 index 0000000..941d088 --- /dev/null +++ b/config.json @@ -0,0 +1,29 @@ +{ + "architectures": [ + "VideoMAEForPreTraining" + ], + "attention_probs_dropout_prob": 0.0, + "decoder_hidden_size": 384, + "decoder_intermediate_size": 1536, + "decoder_num_attention_heads": 6, + "decoder_num_hidden_layers": 4, + "hidden_act": "gelu", + "hidden_dropout_prob": 0.0, + "hidden_size": 768, + "image_size": 224, + "initializer_range": 0.02, + "intermediate_size": 3072, + "layer_norm_eps": 1e-12, + "model_type": "videomae", + "norm_pix_loss": true, + "num_attention_heads": 12, + "num_channels": 3, + "num_frames": 16, + "num_hidden_layers": 12, + "patch_size": 16, + "qkv_bias": true, + "torch_dtype": "float32", + "transformers_version": "4.22.0.dev0", + "tubelet_size": 2, + "use_mean_pooling": false +} diff --git a/model.safetensors b/model.safetensors new file mode 100644 index 0000000..8ecd9b7 --- /dev/null +++ b/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bc053ca2840a038b1068269a4eec06ca569689e9a1ed9376a5b2b8a111be5290 +size 376873760 diff --git a/preprocessor_config.json b/preprocessor_config.json new file mode 100644 index 0000000..631bd2b --- /dev/null +++ b/preprocessor_config.json @@ -0,0 +1,18 @@ +{ + "do_center_crop": true, + "do_normalize": true, + "do_resize": true, + "feature_extractor_type": "VideoMAEFeatureExtractor", + "image_mean": [ + 0.485, + 0.456, + 0.406 + ], + "image_std": [ + 0.229, + 0.224, + 0.225 + ], + "resample": 2, + "size": 224 +} diff --git a/pytorch_model.bin b/pytorch_model.bin new file mode 100644 index 0000000..fbe18aa --- /dev/null +++ b/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2e0c3f4bc73c6d287b96c30975b43d60c9fc004c548150f252278f1becd89a7d +size 376924301