Views
No views yet
pip install modelz
1from modelz import ResnetModel 2 3model = ResnetModel.from_pretrained('nateraw/resnet50') 4ex_input = torch.rand(4, 3, 224, 224) 5out = model(ex_input)