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

torchvision version detection #42

Closed
MamoruDS opened this issue Oct 13, 2021 · 1 comment
Closed

torchvision version detection #42

MamoruDS opened this issue Oct 13, 2021 · 1 comment

Comments

@MamoruDS
Copy link

Hello, I noticed some problems with the implementation of the code that detects the version of torchvision in utils.misc.py

import torchvision
if float(torchvision.__version__[:3]) < 0.7:
    from torchvision.ops import _new_empty_tensor
    from torchvision.ops.misc import _output_size

This will have a fatal issue for torchvision>=0.10

python -c "import torchvision; print(torchvision.__version__[:3])"
# output equals 0.1
jb2020-super added a commit to jb2020-super/Stark that referenced this issue Dec 6, 2021
MasterBin-IIAU added a commit that referenced this issue Jan 6, 2022
fix issue #42, failed when the version of torchvision is '0.10.0'.
@MasterBin-IIAU
Copy link
Collaborator

@MamoruDS Hi, thanks for pointing out this bug. This problem has been fixed.

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

No branches or pull requests

2 participants