Resnet50 Model from Torchvision

Using the model

pip install modelz
from modelz import ResnetModel

model = ResnetModel.from_pretrained('nateraw/resnet50')
ex_input = torch.rand(4, 3, 224, 224)
out = model(ex_input)
Downloads last month
14
Hosted inference API
Image Classification
Drag image file here or click to browse from your device
This model can be loaded on the Inference API on-demand.