我试图在 CTR-GCN 上重新生成结果。给定的代码最初运行良好。然后我将我的 GPU 卡换成了 RTX 3090ti。之后，安装所有 Nvidia 驱动程序。但是现在，当我运行模型时，我收到“警告：根：在输入张量中发现 NaN 或 Inf”这个警告，并使准确性非常低。
如果有人遇到这种情况，请帮我解决这个问题。
当我从 3070 更改为 2060 super 时遇到了完全相同的问题。当我改回 3070 时问题解决了。有趣的是我有另一个模型在 1080ti 上训练，这也不适用于 2060 super 但适用于 3070.