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
我也是照这个方法做的,具体如下: