当前位置:主页 > 软件编程 > Python代码 >

解决pytorch-yolov3 train 报错的问题

时间:2020-10-13 10:06:57 | 栏目:Python代码 | 点击:

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

https://github.com/eriklindernoren/PyTorch-YOLOv3/blob/master/models.py#L191

将model.py 

obj_mask转为int8 bool

obj_mask=obj_mask.bool() # convert int8 to bool

noobj_mask=noobj_mask.bool() #convert int8 to bool 

您可能感兴趣的文章:

相关文章