模型正常进行训练,训练好后想使用一个文本进行一下测试,发现居然在forward函数报错。
开始百思不得其解,训练都是正常的,按evaluate函数的逻辑写了测试样例,怎么会报错呢。
报错如下图:
最后经过仔细的检查,发现是forward()函数里squeeze()函数的问题,这个函数将某些满足只有一个元素的维度删除没了。因此解决办法是使用两个测试样本封装入一个batch中,问题解决了。
还是要更细心
模型正常进行训练,训练好后想使用一个文本进行一下测试,发现居然在forward函数报错。
开始百思不得其解,训练都是正常的,按evaluate函数的逻辑写了测试样例,怎么会报错呢。
报错如下图:
最后经过仔细的检查,发现是forward()函数里squeeze()函数的问题,这个函数将某些满足只有一个元素的维度删除没了。因此解决办法是使用两个测试样本封装入一个batch中,问题解决了。
还是要更细心