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


在用pytorch跑模型时,总是出现一堆这个警告,影响查看模型结果。

在网上找了下解决办法,大多是遵循这篇博客改的:https://blog.csdn.net/BBZZ2/article/details/101022935

将model.py 
obj_mask转为int8 bool
obj_mask=obj_mask.bool() # convert int8 to bool
noobj_mask=noobj_mask.bool() #convert int8 to bool

我也是照这个方法做的,具体如下:


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM