Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Shufflenet v2 pretrained #277

Merged
merged 2 commits into from
Nov 2, 2023
Merged

Shufflenet v2 pretrained #277

merged 2 commits into from
Nov 2, 2023

Conversation

ShawnXuan
Copy link
Collaborator

Added two pre-trained models. These two models are from torchvision, with original files ending in pth. The file names have been modified according to the conventions of flowvision, and they have been compressed into a zip file and uploaded to OSS.

filename from torchvision
shufflenet_v2_x2_0 https://download.pytorch.org/models/shufflenetv2_x2_0-8be3c8ee.pth
shufflenet_v2_x1_5 https://download.pytorch.org/models/shufflenetv2_x1_5-3c479a10.pth

In flowvision/models/utils.py, these two lines will create an empty directory, which will cause the failure of decompression. Therefore, I have removed these two lines.

        if not os.path.exists(extracted_file):
            os.mkdir(extracted_file)

@ShawnXuan ShawnXuan merged commit af65737 into main Nov 2, 2023
0 of 4 checks passed
@ShawnXuan ShawnXuan deleted the shufflenet_v2_pretrained branch November 2, 2023 00:49
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.

3 participants