pytorch - 断言批处理不在键中并且 ptr 不在键中

Question

我正在运行用 PyTorch Lightning 包裹的 PyTorch Geometric (1.7.2)，我得到了上面的断言错误。我无法复制整个堆栈跟踪，但错误是由

return self.collate_fn(data)

我看到这篇文章有同样的错误，但它没有帮助。我将不胜感激有关如何解决此问题的任何提示。