模型正常进行训练，训练好后想使用一个文本进行一下测试，发现居然在forward函数报错。
开始百思不得其解，训练都是正常的，按evaluate函数的逻辑写了测试样例，怎么会报错呢。
报错如下图：
最后经过仔细的检查，发现是forward()函数里squeeze()函数的问题，这个函数将某些满足只有一个元素的维度删除没了。因此解决办法是使用两个测试样本封装入一个batch中，问题解决了。
还是要更细心
![pytorch Dimension out of range (expected to be in range of [-2, 1], but got 2)问题记录](http://www.mshxw.com/aiimages/31/308746.png "pytorch Dimension out of range (expected to be in range of [-2, 1], but got 2)问题记录")
模型正常进行训练，训练好后想使用一个文本进行一下测试，发现居然在forward函数报错。
开始百思不得其解，训练都是正常的，按evaluate函数的逻辑写了测试样例，怎么会报错呢。
报错如下图：
最后经过仔细的检查，发现是forward()函数里squeeze()函数的问题，这个函数将某些满足只有一个元素的维度删除没了。因此解决办法是使用两个测试样本封装入一个batch中，问题解决了。
还是要更细心
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
