-
Notifications
You must be signed in to change notification settings - Fork 7k
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
Test some flaky detection models on float64 instead of float32 #7204
Test some flaky detection models on float64 instead of float32 #7204
Conversation
The last 2 remaining failures are classif-related:
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks Nicolas, LGTM! One non-blocking comment below.
test/test_models.py
Outdated
model.eval().to(device=dev) | ||
x = _get_image(input_shape=input_shape, real_image=real_image, device=dev) | ||
model.eval().to(device=dev, dtype=dtype) | ||
x = _get_image(input_shape=input_shape, real_image=real_image, device=dev).to(dtype) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Should we put the dtype
parameter into _get_image
to avoid an additional copy? This is present in #7130.
Hey @NicolasHug! You merged this PR, but no labels were added. The list of valid labels is available at https://github.com/pytorch/vision/blob/main/.github/process_commit.py |
…32 (#7204) Reviewed By: vmoens Differential Revision: D44416251 fbshipit-source-id: e80c2c1ebde2bad25ab5b6f29e2ac2278e0f4b90
Subset of #7114 to push towards a green[er] CI
Still not ideal at all, but it looks like this makes the detection tests pass again.
cc @pmeier