uoft-cs/cifar100
Viewer • Updated • 60k • 30.7k • 62
This model is a ResNet34 architecture trained from scratch on CIFAR-100 dataset, achieving 73.09% top-1 accuracy.
import torch
from model import ResNet34 # You'll need the model definition
# Load the model
model = ResNet34(num_classes=100)
checkpoint = torch.load('best_model.pth', map_location='cpu')
model.load_state_dict(checkpoint['model_state_dict'])
model.eval()
# For inference, use the transforms from the training
The model can classify images into 100 classes from CIFAR-100 dataset, including animals, vehicles, objects, and natural scenes.