Hemgg commited on
Commit
85f0ae2
·
verified ·
1 Parent(s): 44325d3

End of training

Browse files
Files changed (5) hide show
  1. README.md +62 -0
  2. config.json +38 -0
  3. model.safetensors +3 -0
  4. preprocessor_config.json +26 -0
  5. training_args.bin +3 -0
README.md ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: cc-by-nc-4.0
4
+ base_model: MCG-NJU/videomae-base-finetuned-kinetics
5
+ tags:
6
+ - generated_from_trainer
7
+ metrics:
8
+ - accuracy
9
+ model-index:
10
+ - name: deepfake_model_Video-MAE
11
+ results: []
12
+ ---
13
+
14
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
+ should probably proofread and complete it, then remove this comment. -->
16
+
17
+ # deepfake_model_Video-MAE
18
+
19
+ This model is a fine-tuned version of [MCG-NJU/videomae-base-finetuned-kinetics](https://huggingface.co/MCG-NJU/videomae-base-finetuned-kinetics) on an unknown dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 0.3743
22
+ - Accuracy: 0.8937
23
+
24
+ ## Model description
25
+
26
+ More information needed
27
+
28
+ ## Intended uses & limitations
29
+
30
+ More information needed
31
+
32
+ ## Training and evaluation data
33
+
34
+ More information needed
35
+
36
+ ## Training procedure
37
+
38
+ ### Training hyperparameters
39
+
40
+ The following hyperparameters were used during training:
41
+ - learning_rate: 0.0005
42
+ - train_batch_size: 4
43
+ - eval_batch_size: 4
44
+ - seed: 42
45
+ - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
46
+ - lr_scheduler_type: linear
47
+ - lr_scheduler_warmup_ratio: 0.1
48
+ - num_epochs: 1
49
+
50
+ ### Training results
51
+
52
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
53
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|
54
+ | 0.1978 | 1.0 | 686 | 0.3743 | 0.8937 |
55
+
56
+
57
+ ### Framework versions
58
+
59
+ - Transformers 4.56.1
60
+ - Pytorch 2.8.0+cu126
61
+ - Datasets 4.0.0
62
+ - Tokenizers 0.22.0
config.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "VideoMAEForVideoClassification"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.0,
6
+ "decoder_hidden_size": 384,
7
+ "decoder_intermediate_size": 1536,
8
+ "decoder_num_attention_heads": 6,
9
+ "decoder_num_hidden_layers": 4,
10
+ "dtype": "float32",
11
+ "hidden_act": "gelu",
12
+ "hidden_dropout_prob": 0.0,
13
+ "hidden_size": 768,
14
+ "id2label": {
15
+ "0": "real",
16
+ "1": "fake"
17
+ },
18
+ "image_size": 224,
19
+ "initializer_range": 0.02,
20
+ "intermediate_size": 3072,
21
+ "label2id": {
22
+ "fake": 1,
23
+ "real": 0
24
+ },
25
+ "layer_norm_eps": 1e-12,
26
+ "model_type": "videomae",
27
+ "norm_pix_loss": false,
28
+ "num_attention_heads": 12,
29
+ "num_channels": 3,
30
+ "num_frames": 16,
31
+ "num_hidden_layers": 12,
32
+ "patch_size": 16,
33
+ "problem_type": "single_label_classification",
34
+ "qkv_bias": true,
35
+ "transformers_version": "4.56.1",
36
+ "tubelet_size": 2,
37
+ "use_mean_pooling": true
38
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b276a62de1dbefd3b67bceeac91184375c1f00aec496b5b4b7d0adf61e1a208f
3
+ size 344937328
preprocessor_config.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "crop_size": {
3
+ "height": 224,
4
+ "width": 224
5
+ },
6
+ "do_center_crop": true,
7
+ "do_normalize": true,
8
+ "do_rescale": true,
9
+ "do_resize": true,
10
+ "image_mean": [
11
+ 0.485,
12
+ 0.456,
13
+ 0.406
14
+ ],
15
+ "image_processor_type": "VideoMAEImageProcessor",
16
+ "image_std": [
17
+ 0.229,
18
+ 0.224,
19
+ 0.225
20
+ ],
21
+ "resample": 2,
22
+ "rescale_factor": 0.00392156862745098,
23
+ "size": {
24
+ "shortest_edge": 224
25
+ }
26
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c8dbc54a8c36bb0d8ac544fd7ef3fed032cb82ea3709c89d25a501e9f9f6298d
3
+ size 5777