hub icon indicating copy to clipboard operation
hub copied to clipboard

torch.hub.list error for pytorch/vision

Open addisonklinke opened this issue 4 years ago • 1 comments

Following the documentation example for listing available models, I tried torch.hub.list('pytorch/vision'), but was met with a RuntimeError: builtin cannot be used as a value that appears to originate from the zeros_like() function. I've included the full traceback below

Downloading: "https://github.com/pytorch/vision/archive/master.zip" to /home/addison/.cache/torch/hub/master.zip
Traceback (most recent call last):
  File "/home/addison/miniconda3/envs/tcn/lib/python3.6/site-packages/IPython/core/interactiveshell.py", line 3319, in run_code
    exec(code_obj, self.user_global_ns, self.user_ns)
  File "<ipython-input-4-7cd719e1efcb>", line 1, in <module>
    torch.hub.list('pytorch/vision')
  File "/home/addison/miniconda3/envs/tcn/lib/python3.6/site-packages/torch/hub.py", line 276, in list
    hub_module = import_module(MODULE_HUBCONF, repo_dir + '/' + MODULE_HUBCONF)
  File "/home/addison/miniconda3/envs/tcn/lib/python3.6/site-packages/torch/hub.py", line 72, in import_module
    spec.loader.exec_module(module)
  File "<frozen importlib._bootstrap_external>", line 678, in exec_module
  File "<frozen importlib._bootstrap>", line 219, in _call_with_frames_removed
  File "/home/addison/.cache/torch/hub/pytorch_vision_master/hubconf.py", line 4, in <module>
    from torchvision.models.alexnet import alexnet
  File "/snap/pycharm-community/144/helpers/pydev/_pydev_bundle/pydev_import_hook.py", line 21, in do_import
    module = self._system_import(name, *args, **kwargs)
  File "/home/addison/.cache/torch/hub/pytorch_vision_master/torchvision/__init__.py", line 3, in <module>
    from torchvision import models
  File "/snap/pycharm-community/144/helpers/pydev/_pydev_bundle/pydev_import_hook.py", line 21, in do_import
    module = self._system_import(name, *args, **kwargs)
  File "/home/addison/.cache/torch/hub/pytorch_vision_master/torchvision/models/__init__.py", line 12, in <module>
    from . import detection
  File "/snap/pycharm-community/144/helpers/pydev/_pydev_bundle/pydev_import_hook.py", line 21, in do_import
    module = self._system_import(name, *args, **kwargs)
  File "/home/addison/.cache/torch/hub/pytorch_vision_master/torchvision/models/detection/__init__.py", line 1, in <module>
    from .faster_rcnn import *
  File "/snap/pycharm-community/144/helpers/pydev/_pydev_bundle/pydev_import_hook.py", line 21, in do_import
    module = self._system_import(name, *args, **kwargs)
  File "/home/addison/.cache/torch/hub/pytorch_vision_master/torchvision/models/detection/faster_rcnn.py", line 13, in <module>
    from .rpn import AnchorGenerator, RPNHead, RegionProposalNetwork
  File "/snap/pycharm-community/144/helpers/pydev/_pydev_bundle/pydev_import_hook.py", line 21, in do_import
    module = self._system_import(name, *args, **kwargs)
  File "/home/addison/.cache/torch/hub/pytorch_vision_master/torchvision/models/detection/rpn.py", line 11, in <module>
    from . import _utils as det_utils
  File "/snap/pycharm-community/144/helpers/pydev/_pydev_bundle/pydev_import_hook.py", line 21, in do_import
    module = self._system_import(name, *args, **kwargs)
  File "/home/addison/.cache/torch/hub/pytorch_vision_master/torchvision/models/detection/_utils.py", line 19, in <module>
    class BalancedPositiveNegativeSampler(object):
  File "/home/addison/miniconda3/envs/tcn/lib/python3.6/site-packages/torch/jit/__init__.py", line 1219, in script
    _compile_and_register_class(obj, _rcb, qualified_name)
  File "/home/addison/miniconda3/envs/tcn/lib/python3.6/site-packages/torch/jit/__init__.py", line 1076, in _compile_and_register_class
    _jit_script_class_compile(qualified_name, ast, rcb)
  File "/home/addison/miniconda3/envs/tcn/lib/python3.6/site-packages/torch/jit/_recursive.py", line 222, in try_compile_fn
    return torch.jit.script(fn, _rcb=rcb)
  File "/home/addison/miniconda3/envs/tcn/lib/python3.6/site-packages/torch/jit/__init__.py", line 1226, in script
    fn = torch._C._jit_script_compile(qualified_name, ast, _rcb, get_default_args(obj))
RuntimeError: 
builtin cannot be used as a value:
at /home/addison/.cache/torch/hub/pytorch_vision_master/torchvision/models/detection/_utils.py:14:56
def zeros_like(tensor, dtype):
    # type: (Tensor, int) -> Tensor
    return torch.zeros_like(tensor, dtype=dtype, layout=tensor.layout,
                                                        ~~~~~~~~~~~~~ <--- HERE
                            device=tensor.device, pin_memory=tensor.is_pinned())
'zeros_like' is being compiled since it was called from '__torch__.torchvision.models.detection._utils.BalancedPositiveNegativeSampler.__call__'
at /home/addison/.cache/torch/hub/pytorch_vision_master/torchvision/models/detection/_utils.py:72:12
            # randomly select positive and negative examples
            perm1 = torch.randperm(positive.numel(), device=positive.device)[:num_pos]
            perm2 = torch.randperm(negative.numel(), device=negative.device)[:num_neg]
            pos_idx_per_image = positive[perm1]
            neg_idx_per_image = negative[perm2]
            # create binary mask from indices
            pos_idx_per_image_mask = zeros_like(
            ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~...  <--- HERE
                matched_idxs_per_image, dtype=torch.uint8
            )
            neg_idx_per_image_mask = zeros_like(
                matched_idxs_per_image, dtype=torch.uint8
            )
            pos_idx_per_image_mask[pos_idx_per_image] = torch.tensor(1, dtype=torch.uint8)
            neg_idx_per_image_mask[neg_idx_per_image] = torch.tensor(1, dtype=torch.uint8)

addisonklinke avatar Jan 17 '20 17:01 addisonklinke

Hi @addisonklinke, would you mind trying again with torch version >=1.3.0? Hub always keeps up with the latest pytorch release. I just tried locally and it worked for me.

ailzhang avatar Jan 17 '20 19:01 ailzhang