今天用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
猜你喜欢
转载自blog.csdn.net/weixin_43570254/article/details/104147559
今日推荐
周排行