Cifar 10 Mobilenet

--- license: mit language: - en library_name: pytorch pipeline_tag: image-classification tags: - pytorch - mobilenetv2 - cnn - cifar10 - image-classification - computer-vision datasets: - cifar10 --- # MobileNetV2 on CIFAR-10 A PyTorch implementation of MobileNetV2 trained from scratch on the CIFAR-10 dataset. ## Model Details - Architecture: MobileNetV2 - Framework: PyTorch - Dataset: CIFAR-10 - Input Size: 3 × 32 × 32 - Classes: 10 ## CIFAR-10 Classes | Label | Class | |------:|--------| | 0 | airplane | | 1 | automobile | | 2 | bird | | 3 | cat | | 4 | deer | | 5 | dog | | 6 | frog | | 7 | horse | | 8 | ship | | 9 | truck | ## Training - Optimizer: SGD - Learning Rate: 0.1 - Momentum: 0.9 - Weight Decay: 5e-4 - Scheduler: StepLR - Loss: CrossEntropyLoss - Epochs: 30 - Batch Size: 128 ## Performance | Metric | Value | |--------|------:| | Test Accuracy | 84.62% | ## Model Files - `mobilenetv2_cifar10.pth` ## Load Model ```python model = MobileNetV2() model.load_state_dict( torch.load("mobilenetv2_cifar10.pth") ) model.eval() ``` ## Inference ```python with torch.no_grad(): outputs = model(images) _, predicted = torch.max(outputs, 1) ``` ## Author Ankit Bari - GitHub: https://github.com/aijadugar - Hugging Face: https://huggingface.co/aijadugar

Source context: 11 downloads · 1 likes · Pipeline image-classification · Library pytorch · Repo aijadugar/cifar-10-mobilenet

Source context: 14013 downloads · 1 likes · Pipeline image-classification · Library pytorch · Repo aijadugar/cifar-10-mobilenet