在用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
我也是照這個方法做的,具體如下: