- 博客(1)
- 收藏
- 关注
RSS订阅原创 RuntimeError: Input type (torch.FloatTensor) and weight type (torch.HalfTensor) should be the same
在改写yolox网络架构过程中出现的问题，错误内容大概就是指输入类型是GPU，而权重类型是CPU。当时一直在找weight的加载方式，根据。最后发现是本文改写yolox的部分在于yolo_pafpn.py中，对neck部分的卷积方式做了调整，此处还加上了bias=False，因为跑的时候bias显示也类型不符合，所以就加上了.
2024-04-28 14:51:33
364
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人