koya1 commited on
Commit
5d24664
1 Parent(s): 234d73c

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -9
README.md CHANGED
@@ -16,8 +16,8 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [MCG-NJU/videomae-base](https://huggingface.co/MCG-NJU/videomae-base) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 0.4405
20
- - Accuracy: 0.8903
21
 
22
  ## Model description
23
 
@@ -43,21 +43,21 @@ The following hyperparameters were used during training:
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
  - lr_scheduler_type: linear
45
  - lr_scheduler_warmup_ratio: 0.1
46
- - training_steps: 1200
47
 
48
  ### Training results
49
 
50
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
51
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
52
- | 1.7475 | 0.25 | 300 | 1.1842 | 0.5 |
53
- | 1.5059 | 1.25 | 600 | 1.4309 | 0.6143 |
54
- | 0.1009 | 2.25 | 900 | 1.0795 | 0.8 |
55
- | 1.2366 | 3.25 | 1200 | 0.2548 | 0.9429 |
56
 
57
 
58
  ### Framework versions
59
 
60
- - Transformers 4.26.1
61
  - Pytorch 1.13.1+cu116
62
- - Datasets 2.10.0
63
  - Tokenizers 0.13.2
 
16
 
17
  This model is a fine-tuned version of [MCG-NJU/videomae-base](https://huggingface.co/MCG-NJU/videomae-base) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 0.7549
20
+ - Accuracy: 0.7716
21
 
22
  ## Model description
23
 
 
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
  - lr_scheduler_type: linear
45
  - lr_scheduler_warmup_ratio: 0.1
46
+ - training_steps: 1340
47
 
48
  ### Training results
49
 
50
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
51
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
52
+ | 1.2417 | 0.25 | 335 | 1.5928 | 0.3514 |
53
+ | 0.7873 | 1.25 | 670 | 1.0957 | 0.5 |
54
+ | 0.7945 | 2.25 | 1005 | 1.0748 | 0.7027 |
55
+ | 0.0604 | 3.25 | 1340 | 0.7425 | 0.7297 |
56
 
57
 
58
  ### Framework versions
59
 
60
+ - Transformers 4.27.3
61
  - Pytorch 1.13.1+cu116
62
+ - Datasets 2.10.1
63
  - Tokenizers 0.13.2