在用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

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

版权声明:本文为zhibei原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://www.cnblogs.com/zhibei/p/12106893.html