{ "_name_or_path": "MCG-NJU/videomae-base", "architectures": [ "VideoMAEForVideoClassification" ], "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, "id2label": { "0": "bending_bad", "1": "bending_good", "2": "carry_n_walk_bad", "3": "carry_n_walk_good", "4": "lifting_bad", "5": "lifting_good", "6": "pull_bad", "7": "pull_good", "8": "push_bad", "9": "push_good", "10": "twist_bad", "11": "twist_good" }, "image_size": 224, "initializer_range": 0.02, "intermediate_size": 3072, "label2id": { "bending_bad": 0, "bending_good": 1, "carry_n_walk_bad": 2, "carry_n_walk_good": 3, "lifting_bad": 4, "lifting_good": 5, "pull_bad": 6, "pull_good": 7, "push_bad": 8, "push_good": 9, "twist_bad": 10, "twist_good": 11 }, "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, "problem_type": "single_label_classification", "qkv_bias": true, "torch_dtype": "float32", "transformers_version": "4.29.1", "tubelet_size": 2, "use_mean_pooling": false }