detr-resnet-50_finetuned_cppe5
talk2raja
Detección de objetos
Este modelo es una versión afinada de facebook/detr-resnet-50 en el conjunto de datos cppe-5.
Como usar
El siguiente conjunto de hiperparámetros fue utilizado durante el entrenamiento:
- Tasa de aprendizaje: 1e-05
- Tamaño del lote de entrenamiento: 8
- Tamaño del lote de evaluación: 8
- Semilla: 42
- Optimizador: Adam con betas=(0.9,0.999) y épsilon=1e-08
- Tipo de programador de tasa de aprendizaje: lineal
- Número de épocas: 10
- Entrenamiento de precisión mixta: AMP nativo
Versiones del marco de trabajo utilizadas:
- Transformers 4.28.1
- Pytorch 2.0.0+cu118
- Datasets 2.12.0
- Tokenizers 0.13.3
El modelo todavía no tiene suficiente actividad para ser desplegado en la API de inferencia (sin servidor). Aumenta su visibilidad social y verifica más tarde, o desplégalo en Puntos de Inferecia (dedicados) en su lugar.
Funcionalidades
- Detección de objetos
- Transformers
- PyTorch
- Integración con TensorBoard
Casos de uso
- Detección de objetos en imágenes