网络编程
位置:首页>> 网络编程>> 网络编程>> 解决pytorch-yolov3 train 报错的问题

解决pytorch-yolov3 train 报错的问题

作者:bbzz2  发布时间:2023-11-26 04:14:29 

标签:pytorch,yolov3,train

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

解决pytorch-yolov3 train 报错的问题

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

来源:https://blog.csdn.net/BBZZ2/article/details/101022935

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com