hook-torchvision.py 750 B

123456789101112131415161718
  1. # ------------------------------------------------------------------
  2. # Copyright (c) 2020 PyInstaller Development Team.
  3. #
  4. # This file is distributed under the terms of the GNU General Public
  5. # License (version 2.0 or later).
  6. #
  7. # The full license is available in LICENSE, distributed with
  8. # this software.
  9. #
  10. # SPDX-License-Identifier: GPL-2.0-or-later
  11. # ------------------------------------------------------------------
  12. # Functions from torchvision.ops.* modules require torchvision._C extension module, which PyInstaller fails to pick up
  13. # automatically due to indirect load.
  14. hiddenimports = ['torchvision._C']
  15. # Collect source .py files for JIT/torchscript. Requires PyInstaller >= 5.3, no-op in older versions.
  16. module_collection_mode = 'pyz+py'