hafidber
/

videomae-base-finetuned-ucf101-subset

Video Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

hafidber commited on Oct 17, 2023

Commit

a67d0d2

·

1 Parent(s): 60446bf

Model save

Files changed (1) hide show

README.md +8 -4

README.md CHANGED Viewed

@@ -17,8 +17,8 @@ should probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [MCG-NJU/videomae-base](https://huggingface.co/MCG-NJU/videomae-base) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.4294
-- Accuracy: 0.5290
 ## Model description
@@ -44,13 +44,17 @@ The following hyperparameters were used during training:
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_ratio: 0.1
-- training_steps: 150
 ### Training results
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 1.3265        | 1.0   | 150  | 1.2609          | 0.6286   |
 ### Framework versions

 This model is a fine-tuned version of [MCG-NJU/videomae-base](https://huggingface.co/MCG-NJU/videomae-base) on an unknown dataset.
 It achieves the following results on the evaluation set:
+- Loss: 0.3160
+- Accuracy: 0.8968
 ## Model description
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_ratio: 0.1
+- training_steps: 750
 ### Training results
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
+| 1.5236        | 0.2   | 150  | 1.5999          | 0.3143   |
+| 0.7562        | 1.2   | 300  | 0.7772          | 0.7571   |
+| 0.1744        | 2.2   | 450  | 0.6027          | 0.8286   |
+| 0.6707        | 3.2   | 600  | 0.4752          | 0.8286   |
+| 0.0057        | 4.2   | 750  | 0.0703          | 0.9714   |
 ### Framework versions