2301_78886263的博客

原创 INTERNAL ASSERT FAILED

在pytorch2.3.0和pytorch2.1.2里报了相同的错错误，一开始以为是显存不够造成的，尝试了调小batch_size，修改gru隐藏节点，修改滤波器大小等方法都没有解决。都没有成功，于是我开始排查自己对代码所做的改动，最大的改动是注意力模块，而其中最有可能出问题的，我认为是。试了搜到的很多种办法，包括csdn里其他朋友的办法以及GitHub里的办法。，自适应平均池化，将其改为对应的平均池化之后，我的问题就解决了，没再出现。3.换回pytorch1.x.x版本。第一次遇到这个报错记录一下。