Skip to content

Add hubconf.py and automatic downloading of some pre-trained parameters#29

Open
Ivan1248 wants to merge 8 commits intoJIA-Lab-research:masterfrom
Ivan1248:master
Open

Add hubconf.py and automatic downloading of some pre-trained parameters#29
Ivan1248 wants to merge 8 commits intoJIA-Lab-research:masterfrom
Ivan1248:master

Conversation

@Ivan1248
Copy link
Copy Markdown

@Ivan1248 Ivan1248 commented Feb 28, 2022

This adds hubconf.py so that, for example, the following can work when "Ivan1248" is replaced with "dvlab-research":

model = torch.hub.load('Ivan1248/Context-Aware-Consistency', 'DeepLabV3Plus', backbone='resnet50',
                       pretrained=True, num_classes=21)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant