如何使用一个模型输入多个数据？

Question

我正在训练文档分类器，并且想要使用辅助损失。但是，当模型获得输入时，总是导致CUDA内存不足。

这是伯特模型。我确定该实验已经足够了。

ViewModel

遇到此错误时，请获取此错误“ RuntimeError：CUDA内存不足”。 有什么建议吗？谢谢。