登录社区云,与社区用户共同成长
邀请您加入社区
该用户还未填写简介
暂无可提供的服务
在初始化模型时,传入id2label和label2id可以用符合当前任务的分类头替换原来的分类头,如果只传num_labels也是可以的,不过后续模型输出需要自己写个映射。定义TrainingArgumentsargs=args,