Skip to content

November Build

Compare
Choose a tag to compare
@Legorooj Legorooj released this 21 Dec 07:56
· 671 commits to master since this release
861fbf6
Add a hook for torchvision.ops (#80)

* tests: add a test for torchvision.ops.nms

Reproduces the error from pyinstaller/pyinstaller#5386.

* Add a hook for torchvision.ops

Add a hidden import of torchvision._C extension module that we
fail to pick up automatically. This extension module is required
in functions from torchvision.ops.* modules, where its absence
causes the _assert_has_ops() call to fail.

Fixes the failing test_torchvision_nms.
Fixes pyinstaller/pyinstaller#5386.