Retrosynthesis-Prediction icon indicating copy to clipboard operation
Retrosynthesis-Prediction copied to clipboard

Error in utils.py

Open QuickLearner171998 opened this issue 5 years ago • 0 comments

Hi, I tried running the Retrosynthesis Reaction Planning.ipynb notebook. I am getting this error:

RuntimeError: builtin cannot be used as a value: at /usr/local/lib/python3.6/dist-packages/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 /usr/local/lib/python3.6/dist-packages/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)

QuickLearner171998 avatar Feb 13 '20 05:02 QuickLearner171998