Dmitry comments

Repositories
Issues
Comments

Results 1 comments of


                                            Dmitry

UserWarning: indexing with dtype torch.uint8 is now deprecated, please use a dtype torch.bool instead.

Replace in [utils:269](https://github.com/eriklindernoren/PyTorch-YOLOv3/blob/47b7c912877ca69db35b8af3a38d6522681b3bb3/utils/utils.py#L269) `ByteTensor = torch.cuda.ByteTensor if pred_boxes.is_cuda else torch.ByteTensor` With: `BoolTensor = torch.cuda.BoolTensor if pred_boxes.is_cuda else torch.BoolTensor` And it usage in lines 278, 279 `obj_mask = BoolTensor(nB, nA, nG,...