开发者社区 > ModelScope模型即服务 > 计算机视觉 > 正文

我要在ModelScope自有的数据上finetune CoROM模型,我的这个理解正确吗?

我要在ModelScope自有的数据上finetune CoROM模型,cfg中dataset部分修改如下:
af97bd572c0edddf903df455561709e6.png
代码执行到train_loop的位置发现其使用了TextRankingDataset
3515f009307a35b9f40d1d2f26ed91bb.png
我看了dataset部分的代码,没有发现与“first_sequence、second_sequence”适配的dataset。
目前看来我需根据自己的数据格式实现一个类似TextRankingDataset的Dataset。
我的问题是:这个理解正确吗?

展开
收起
2401。 2023-09-02 20:22:58 67 0
1 条回答
写回答
取消 提交回答
  • 可以参考ds = MsDataset.load('msmarco-passage-ranking', 'zyznull') 这个数据集的格式自己上传一个数据集训练。
    此回答整理自“魔搭ModelScope开发者联盟群 ①”。

    2023-09-05 12:16:18
    赞同 展开评论 打赏

包含图像分类、图像生成、人体人脸识别、动作识别、目标分割、视频生成、卡通画、视觉评价、三维视觉等多个领域

相关电子书

更多
视觉AI能力的开放现状及ModelScope实战 立即下载
ModelScope助力语音AI模型创新与应用 立即下载
低代码开发师(初级)实战教程 立即下载