训练yolox的时候遇到这个问题。 解决方法 原先代码 yv, xv = torch.meshgrid([torch.arange(hsize), torch.arange(wsize)], indexing="ij") 删除indexing="ij"即可 yv, xv = torch.meshgrid([torch.arange(hsize), torch.arange(wsize)])
在训练Yolox模型时遇到了一个问题，原代码在创建meshgrid时指定了`indexing=ij`，这导致了问题。通过删除这一参数，即`yv,xv=torch.meshgrid([torch.arange(hsize),torch.arange(wsize)])`，问题得到了解决。这个改动对于理解PyTorch中网格生成的细节以及如何避免此类错误是有帮助的。
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
