torch meshgrid 警告：在即将发布的版本中，将需要传递索引参数

Question

我尝试执行 LIIF(https://github.com/yinboc/liif) 并出现以下警告：

/usr/local/lib/python3.7/dist-packages/torch/functional.py:445: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at  ../aten/src/ATen/native/TensorShape.cpp:2157.)
  return _VF.meshgrid(tensors, **kwargs)  # type: ignore[attr-defined]

Answer 1

更改为

return _VF.meshgrid(tensors, **kwargs, indexing="ij")

Answer 2

我的答案可能不是直接的解决方案，但可能有一定的相关性。我遇到了同样的警告，但它是由调用

torch.cartesian_prod(*tensors)

引起的。

/opt/conda/lib/python3.7/site-packages/torch/functional.py:1069: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at  /opt/conda/conda-bld/pytorch_1634272168290/work/aten/src/ATen/native/TensorShape.cpp:2157.)return _VF.cartesian_prod(tensors)  # type: ignore[attr-defined]

从我个人的经验来看，我找不到

_VF.cartesian_prod

和

meshgrid

之间有任何亲戚，但警告是存在的。其他一些人也遇到过这个问题，请参阅pytorch问题#50276。 Pytorch 的工程师承诺会看一下，但我猜他们可能会忽略（或者可能忘记）这一点。

所以我相信这可能不是一个严重的问题并阻碍 Pytorch 的使用。继续吧，看看在未来的版本中是否需要任何新参数（例如索引）。

Answer 3

我将 pytorch 版本更改为 22.10 nvcr.io/nvidia/pytorch:22.10-py3，它有助于解决问题。它是在我训练yolov8时出现的。