-
Nicolas Hug authored
Summary: Pull Request resolved: https://github.com/pytorch/vision/pull/3748 This PR parametrizes the `perspective`-related tests, and avoids having deeply nested for-loops which will help debugging. "What" gets tested is left unchanged. The newly-introduced `cpu_and_gpu()` generator along with the `dont_collect` mark is a logic that allows to not run CPU tests on GPU machines (and vice versa). Reviewed By: fmassa Differential Revision: D27908299 fbshipit-source-id: 24a10a89fe90ae0a9e62de4bc7e768a669ebf212
1b0bd0e3
