MobileNetV3 Small — CIFAR-10
Model torchvision z wagami ImageNet, dostrojony na CIFAR-10.
Trening
- 5000 obrazów treningowych
- 1000 obrazów walidacyjnych
- 2 epoki, batch size 64, AdamW, learning rate 0.001
- Seed: 42
- Najlepszy checkpoint wybrany według walidacyjnego accuracy
- Oddzielny oficjalny test CIFAR-10: 10000 obrazów
Wyniki
- Best validation accuracy: 0.7320
- Test accuracy: 0.7232
- Test macro F1: 0.7204
Wczytanie
Utwórz torchvision.models.mobilenet_v3_small(weights=None). Zastąp model.classifier[-1] warstwą Linear o 10 wyjściach. Wczytaj state_dict z model.pth i wywołaj model.eval(). Kolejność klas znajduje się w config.json. Preprocessing: MobileNet_V3_Small_Weights.IMAGENET1K_V1.transforms().
Ograniczenia
Projekt edukacyjny, krótki trening. Model rozpoznaje tylko klasy CIFAR-10. Wyniki na innych rodzajach zdjęć mogą być słabsze.