detr-resnet-50_finetuned_brain_tumor

mmenendezg
Detección de objetos

Este modelo es una versión afinada de facebook/detr-resnet-50 en el conjunto de datos None.

Como usar

Procedimiento de entrenamiento

Los siguientes hiperparámetros fueron utilizados durante el entrenamiento:

  • learning_rate: 9.764150575493936e-06
  • train_batch_size: 4
  • eval_batch_size: 8
  • seed: 42
  • optimizer: Adam con betas=(0.9,0.999) y epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 50

Resultados de entrenamiento

Versiones del framework

  • Transformers 4.33.2
  • Pytorch 2.0.1
  • Datasets 2.14.5
  • Tokenizers 0.13.3

Funcionalidades

Detección de objetos
Transformers
Compatible con PyTorch
Generado a partir de Trainer

Casos de uso

Detección de tumores cerebrales en imágenes médicas