Retrosynthesis-Prediction
Retrosynthesis-Prediction copied to clipboard
Error in utils.py
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)