Hugging Face's logo

prathameshdalal
/
videomae-base-finetuned-ucf101-subset

Video Classification
Transformers
PyTorch
TensorBoard
videomae
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
prathameshdalal commited on
Commit
dac7da5
·
1 Parent(s): 5b1f5b4

Training in progress, epoch 0

Browse files
Files changed (3) hide show
  1. README.md +9 -21
  2. pytorch_model.bin +1 -1
  3. training_args.bin +1 -1
README.md CHANGED
@@ -17,8 +17,8 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  This model is a fine-tuned version of [MCG-NJU/videomae-base](https://huggingface.co/MCG-NJU/videomae-base) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 0.0480
21
- - Accuracy: 0.9857
22
 
23
  ## Model description
24
 
@@ -38,8 +38,8 @@ More information needed
38
 
39
  The following hyperparameters were used during training:
40
  - learning_rate: 5e-05
41
- - train_batch_size: 8
42
- - eval_batch_size: 8
43
  - seed: 42
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: linear
@@ -50,27 +50,15 @@ The following hyperparameters were used during training:
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
52
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
53
- | 2.2995 | 0.06 | 38 | 2.1143 | 0.3 |
54
- | 1.6428 | 1.06 | 76 | 1.5030 | 0.3429 |
55
- | 0.6261 | 2.06 | 114 | 0.8121 | 0.6286 |
56
- | 0.4044 | 3.06 | 152 | 0.4910 | 0.7714 |
57
- | 0.3236 | 4.06 | 190 | 0.4558 | 0.8714 |
58
- | 0.1865 | 5.06 | 228 | 0.6773 | 0.8 |
59
- | 0.1479 | 6.06 | 266 | 0.1813 | 0.9571 |
60
- | 0.0418 | 7.06 | 304 | 0.1312 | 0.9714 |
61
- | 0.0718 | 8.06 | 342 | 0.2473 | 0.9286 |
62
- | 0.0596 | 9.06 | 380 | 0.1378 | 0.9714 |
63
- | 0.0079 | 10.06 | 418 | 0.0308 | 0.9857 |
64
- | 0.0134 | 11.06 | 456 | 0.0329 | 1.0 |
65
- | 0.0083 | 12.06 | 494 | 0.0310 | 1.0 |
66
- | 0.0394 | 13.06 | 532 | 0.0543 | 0.9714 |
67
- | 0.004 | 14.06 | 570 | 0.0492 | 0.9857 |
68
- | 0.0043 | 15.05 | 600 | 0.0480 | 0.9857 |
69
 
70
 
71
  ### Framework versions
72
 
73
- - Transformers 4.31.0
74
  - Pytorch 1.10.0+cu113
75
  - Datasets 2.14.4
76
  - Tokenizers 0.13.3
 
17
 
18
  This model is a fine-tuned version of [MCG-NJU/videomae-base](https://huggingface.co/MCG-NJU/videomae-base) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 0.2538
21
+ - Accuracy: 0.9
22
 
23
  ## Model description
24
 
 
38
 
39
  The following hyperparameters were used during training:
40
  - learning_rate: 5e-05
41
+ - train_batch_size: 2
42
+ - eval_batch_size: 2
43
  - seed: 42
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: linear
 
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
52
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
53
+ | 1.5736 | 0.25 | 150 | 1.4544 | 0.4429 |
54
+ | 0.6422 | 1.25 | 300 | 0.6985 | 0.6857 |
55
+ | 0.0978 | 2.25 | 450 | 0.2436 | 0.8857 |
56
+ | 0.2305 | 3.25 | 600 | 0.2538 | 0.9 |
 
 
 
 
 
 
 
 
 
 
 
 
57
 
58
 
59
  ### Framework versions
60
 
61
+ - Transformers 4.32.1
62
  - Pytorch 1.10.0+cu113
63
  - Datasets 2.14.4
64
  - Tokenizers 0.13.3
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:50df48695592c0936e92312e527458b8475f7adb703943cbb20ef7b17af58717
3
  size 344999245
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7897dd1ce22105d5dae5eb3af6470d181281de213c04938307969930c4978e12
3
  size 344999245
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f49f0b2156d66fb674450ccff1180257785f4f60220a346074020e9dd8353547
3
  size 4079
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8f07eb5708ebf6d25363ea8e54b0c0408e958be7b90dee6cf1ffbfe328e92355
3
  size 4079