mnist-mlp

dacorvo

Clasificación de imagen

Este es un MLP simple entrenado en el conjunto de datos MNIST. Su uso principal es ser un modelo de referencia muy simple para probar la cuantización.

Como usar

Para usar este modelo, las imágenes MNIST deben ser normalizadas y aplanadas de la siguiente manera:
from torchvision import datasets, transforms

transform=transforms.Compose([
    transforms.ToTensor(),
    transforms.Normalize((0.1307,), (0.3081,)),
    transforms.Lambda(lambda x: torch.flatten(x)),
])
test_set = datasets.MNIST('../data', train=False, download=True, transform=transform)

Funcionalidades

Clasificación de imágenes
Transformers
PyTorch
Extracción de características
pre-entrenado
código personalizado

Casos de uso

Modelo de referencia para probar la cuantización