videomae-base-finetuned-fight-nofight-subset

archit11
Clasificación de video

Este modelo es una versión ajustada de MCG-NJU/videomae-base en un conjunto de datos desconocido. Se utiliza para detectar peleas en grabaciones de CCTV.

Como usar

Procedimiento de entrenamiento

Hiperparámetros de entrenamiento

Los siguientes hiperparámetros se utilizaron durante el entrenamiento:

  • learning_rate: 5e-05
  • train_batch_size: 4
  • eval_batch_size: 4
  • seed: 42
  • optimizer: Adam con betas=(0.9,0.999) y epsilon=1e-08
  • lr_scheduler_type: linear
  • lr_scheduler_warmup_ratio: 0.1
  • training_steps: 102

Versiones de los frameworks

  • Transformers: 4.35.2
  • Pytorch: 2.1.0+cu121
  • Datasets: 2.16.1
  • Tokenizers: 0.15.1

Funcionalidades

Clasificación de video
Transformers
TensorBoard
Safetensors

Casos de uso

Detectar peleas en grabaciones de CCTV