如何解决pytorch-yolov3train报错的问题-创新互联

这篇文章主要介绍了如何解决pytorch-yolov3 train报错的问题,具有一定借鉴价值,感兴趣的朋友可以参考下,希望大家阅读完这篇文章之后大有收获,下面让小编带着大家一起了解一下。

成都创新互联主要从事网站设计、成都网站制作、网页设计、企业做网站、公司建网站等业务。立足成都服务临潭,10多年网站建设经验,价格优惠、服务专业,欢迎来电咨询建站服务:18980820575

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

感谢你能够认真阅读完这篇文章,希望小编分享的“如何解决pytorch-yolov3 train报错的问题”这篇文章对大家有帮助,同时也希望大家多多支持创新互联成都网站设计公司,关注创新互联成都网站设计公司行业资讯频道,更多相关知识等着你来学习!

另外有需要云服务器可以了解下创新互联scvps.cn,海内外云服务器15元起步,三天无理由+7*72小时售后在线,公司持有idc许可证,提供“云服务器、裸金属服务器、网站设计器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案,具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势,专为企业上云打造定制,能够满足用户丰富、多元化的应用场景需求。


当前名称:如何解决pytorch-yolov3train报错的问题-创新互联
本文来源:http://abwzjs.com/article/jhedc.html