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

Version check (torchvision) >= 0.10 error fix #7

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

EherSenaw
Copy link

@EherSenaw EherSenaw commented Apr 8, 2022

TL;DR ) Fixed util/misc.py's version checking method for torchvision to be compatible with torchvision >= 0.10.0

Since I was interested in your paper, Sparse DETR, I was trying to run(train/test) model within Google Colab environment.

During the initial run, I found out that torchvision version checking method does not work as intended:

  • My torchvision version was0.10.1, but the checking method treated as 0.1.

Solved by replacing

float(torchvision.__version__[:3]) < 0.5

with

float(torchvision.__version__.split('.')[1]) < 5

And for <0.7, similarly fixed.

Therefore, I request pull-request to fix this issue.

Thank you.
Taewoo Jeong.

@EherSenaw EherSenaw changed the title Version check (torchvision) >= 0.10 error fixed Version check (torchvision) >= 0.10 error fix Apr 8, 2022
@EherSenaw EherSenaw closed this Apr 8, 2022
@EherSenaw EherSenaw reopened this Apr 8, 2022
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