Model save

README.md

@@ -3,6 +3,8 @@ license: cc-by-nc-4.0
 base_model: MCG-NJU/videomae-base
 tags:
 - generated_from_trainer
+metrics:
+- accuracy
 model-index:
 - name: videomae-base-finetuned-ucf101-subset
   results: []
@@ -14,6 +16,9 @@ should probably proofread and complete it, then remove this comment. -->
 # videomae-base-finetuned-ucf101-subset
 
 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.5321
+- Accuracy: 0.8645
 
 ## Model description
 
@@ -41,6 +46,16 @@ The following hyperparameters were used during training:
 - lr_scheduler_warmup_ratio: 0.1
 - training_steps: 148
 
+### Training results
+
+| Training Loss | Epoch  | Step | Validation Loss | Accuracy |
+|:-------------:|:------:|:----:|:---------------:|:--------:|
+| 2.1431        | 0.2568 | 38   | 1.7960          | 0.5857   |
+| 0.9034        | 1.2568 | 76   | 0.9433          | 0.7571   |
+| 0.4889        | 2.2568 | 114  | 0.5177          | 0.8857   |
+| 0.3076        | 3.2297 | 148  | 0.4174          | 0.9143   |
+
+
 ### Framework versions
 
 - Transformers 4.41.2