license: mit language: en library name: pytorch pipeline tag: image classification tags: pytorch resnet18 cnn cifar10 image classification computer vision datasets: cifar10 ResNet 18 on CIFAR 10 A PyTorch implementation of ResNet 18 trained from scratch on the CIFAR 10 dataset. Model Details Architecture: ResNet 18 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 89.61% Model Files resnet18 cifar10.pth Load Model Inference Author Ankit Bari GitHub: https://github.com/aijadugar Hugging Face: https://huggingface.co/aijadugar
We use cookies for essential functionality and analytics. You can accept or reject analytics cookies.Cookie policy