zhibei

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

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

分类:

技术点:

相关文章: