用大语言模型做文本分类任务
遇到以下的问题:
首先我的dataset类如下:
将文本(text)处理成长度512的张量,label为整数(0,1)
但是用dataloader训练模型的报错信息如下:
求大佬解决
如果有用大语言模型做文本分类任务的项目也请告知我,谢谢~~
用大语言模型做文本分类任务
遇到以下的问题:
首先我的dataset类如下:
将文本(text)处理成长度512的张量,label为整数(0,1)
但是用dataloader训练模型的报错信息如下:
求大佬解决
如果有用大语言模型做文本分类任务的项目也请告知我,谢谢~~