Skip to content

GitLab

"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "d27e996ccdecbddaf37f1c41a9d0a0e6fabb4fbf"
Commit 11821d62 authored by Sam Gross's avatar Sam Gross
Browse files

Add ResNet-101 pre-trained model

parent 7d150e8f
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 2 deletions
torchvision/models/resnet.py
View file @ 11821d62
...@@ -11,6 +11,7 @@ model_urls = { ...@@ -11,6 +11,7 @@ model_urls = {
'resnet18': 'https://s3.amazonaws.com/pytorch/models/resnet18-5c106cde.pth', 'resnet18': 'https://s3.amazonaws.com/pytorch/models/resnet18-5c106cde.pth',
'resnet34': 'https://s3.amazonaws.com/pytorch/models/resnet34-333f7ec4.pth', 'resnet34': 'https://s3.amazonaws.com/pytorch/models/resnet34-333f7ec4.pth',
'resnet50': 'https://s3.amazonaws.com/pytorch/models/resnet50-19c8e357.pth', 'resnet50': 'https://s3.amazonaws.com/pytorch/models/resnet50-19c8e357.pth',
'resnet101': 'https://s3.amazonaws.com/pytorch/models/resnet101-5d3b4d8f.pth',
} }
...@@ -171,8 +172,11 @@ def resnet50(pretrained=False): ...@@ -171,8 +172,11 @@ def resnet50(pretrained=False):
return model return model
def resnet101(): def resnet101(pretrained=False):
return ResNet(Bottleneck, [3, 4, 23, 3]) model = ResNet(Bottleneck, [3, 4, 23, 3])
if pretrained:
model.load_state_dict(model_zoo.load_url(model_urls['resnet101']))
return model
def resnet152(): def resnet152():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment