Model save

README.md

@@ -18,7 +18,7 @@ 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: 0.2473
+- Loss: 0.0029
 - Accuracy: 1.0
 
 ## Model description
@@ -39,21 +39,22 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 5e-05
-- train_batch_size: 16
-- eval_batch_size: 16
+- train_batch_size: 8
+- eval_batch_size: 8
 - seed: 42
 - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_ratio: 0.1
-- training_steps: 12
+- training_steps: 48
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Accuracy |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|
-| No log        | 1.0   | 4    | 0.2036          | 1.0      |
-| No log        | 2.0   | 8    | 0.0109          | 1.0      |
-| 0.4029        | 3.0   | 12   | 0.0042          | 1.0      |
+| Training Loss | Epoch  | Step | Validation Loss | Accuracy |
+|:-------------:|:------:|:----:|:---------------:|:--------:|
+| 0.4706        | 1.0    | 13   | 0.0024          | 1.0      |
+| 0.0027        | 2.0    | 26   | 0.0007          | 1.0      |
+| 0.0008        | 3.0    | 39   | 0.0005          | 1.0      |
+| 0.0005        | 3.6923 | 48   | 0.0005          | 1.0      |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9bba6b2e96d748aa9cb0a45d43995657e13f0e40efbbf69efbf74be36c33fda4
+oid sha256:ae97de9d2f80c1f70886a6f3a2fff46bbfa607423ff450b3059f929f12622b8f
 size 344940444