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