Liu yuyan
Liu yuyan
首先确认你的python库版本正确，如果仍然出现问题，应该是显存。 解决方法一：然后你用自己的模型在中文数据集上32G的显存是不能跑32batch的，去修改Textattack\textattack\models\wrappers\pytorch_model_wrapper.py下的batchsize。 解决方法二：一般你的句子不会太长，textattack默认用了512长度，你可以考虑截断一下，这样会非常省显存： 
This is a very valuable question