今天用pytorch训练的时候,出现了一堆这样的报错,翻阅了一些资料,找到了解决方法
找到model.py
将obj_mask改为obj_mask.bool()
将noobj_mask改为noobj——mask.bool()
我在repalce all的时候,发现部分已经加了.bool()所以在替换的时候要注意一些
UserWarning: indexing with dtype torch.uint8 is now deprecated, please use a dtype torch.bool instea
最新推荐文章于 2023-03-24 23:42:39 发布