KingTechnician/videomae-small-finetuned-kinetics-xd-violence-binary

<!-- This model card has been generated automatically according to the information the Trainer had access to. You should probably proofread and complete it, then remove this comment. -->

videomae-small-finetuned-kinetics-finetuned-xd-violence

This model is a fine-tuned version of MCG-NJU/videomae-small-finetuned-kinetics on an unknown dataset. It achieves the following results on the evaluation set:

• —Loss: 0.3828
• —Accuracy: 0.8254

Video preprocessing config: Resize to: (224, 224) Frames to sample: 16 Clip duration: 2.1333333333333333s

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
• —trainbatchsize: 2
• —evalbatchsize: 2
• —seed: 42
• —gradientaccumulationsteps: 4
• —totaltrainbatch_size: 8
• —optimizer: Use OptimizerNames.ADAMWTORCHFUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
• —lrschedulertype: linear
• —lrschedulerwarmup_ratio: 0.1
• —num_epochs: 3

Training results

Testing results

{'eval_loss': 0.3827999532222748,
 'eval_accuracy': 0.8253768844221105,
 'eval_runtime': 184.3164,
 'eval_samples_per_second': 4.319,
 'eval_steps_per_second': 2.159,
 'epoch': 3.0}

Framework versions

• —Transformers 4.57.1
• —Pytorch 2.8.0+cu126
• —Datasets 4.0.0
• —Tokenizers 0.22.1