vivit-vit-fc-loss-8-10-epochs-5_class_1_p0_splited-1788680718.6508312
This model is a fine-tuned version of google/vivit-b-16x2-kinetics400 on an unknown dataset. It achieves the following results on the evaluation set:
- Loss: 1.0742
- Accuracy: 0.7316
Model description
More information needed
Intended uses & limitations
More information needed
Training and evaluation data
More information needed
Training procedure
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 5e-05
- train_batch_size: 16
- eval_batch_size: 16
- seed: 42
- optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
- lr_scheduler_type: linear
- training_steps: 8660
Training results
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|---|---|---|---|---|
| 0.5371 | 0.0501 | 434 | 0.5556 | 0.6344 |
| 0.4241 | 1.0501 | 868 | 0.5612 | 0.6505 |
| 0.3772 | 2.0501 | 1302 | 0.5229 | 0.6701 |
| 0.3835 | 3.0501 | 1736 | 0.6226 | 0.6586 |
| 0.3312 | 4.0501 | 2170 | 0.5428 | 0.7059 |
| 0.2258 | 5.0501 | 2604 | 0.5670 | 0.7243 |
| 0.2561 | 6.0501 | 3038 | 0.5991 | 0.7336 |
| 0.2365 | 7.0501 | 3472 | 0.7801 | 0.7370 |
| 0.2588 | 8.0501 | 3906 | 0.6824 | 0.7301 |
| 0.1235 | 9.0501 | 4340 | 0.8167 | 0.7232 |
| 0.0821 | 10.0501 | 4774 | 0.8405 | 0.7301 |
| 0.1289 | 11.0501 | 5208 | 0.9522 | 0.7059 |
| 0.1667 | 12.0501 | 5642 | 0.9302 | 0.7209 |
| 0.1119 | 13.0501 | 6076 | 1.1207 | 0.7266 |
| 0.0779 | 14.0501 | 6510 | 1.0082 | 0.7209 |
| 0.0660 | 15.0501 | 6944 | 1.1594 | 0.7209 |
| 0.0792 | 16.0501 | 7378 | 1.1923 | 0.7278 |
| 0.1256 | 17.0501 | 7812 | 1.3317 | 0.7151 |
| 0.0637 | 18.0501 | 8246 | 1.2180 | 0.7278 |
| 0.0597 | 19.0478 | 8660 | 1.2229 | 0.7290 |
Framework versions
- Transformers 5.16.1
- Pytorch 2.14.0+cu126
- Datasets 5.0.1
- Tokenizers 0.23.2