ModelScope使用的是本地load用户自定义数据,出现报错?

ModelScope通过官网进行GeoGLUE的Query-POI排序任务的自定义训练,使用的是本地load用户自定义数据,保存为train.json和dev.json,出现报错:
RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 20 but got size 2 for tensor number 2 in the list?damo/mgeo_backbone_chinese_base刚刚应该是未更新原始代码中的neg_sample数量为自定义训练集的负例个数,刚由19改成1了,现在报新的错误,KeyError: 'mrr@1 is not in the metrics registry group default. Please make sure the correct version of ModelScope library is used.',ModelScope版本为1.9.1{'query_id': 0, 'query': '丽华公寓(通惠中路)向北检验检疫科学研究所', 'query_gis': '[[], [], [], [], [], "120.59443087451544,30.315515932852602"]', 'idx': '0', 'positive_passages': [{'text': '杭州中新街(惠港二路)76饶平县检验检疫局', 'gis': '[[], [], [], [], [], "120.20509044775532,30.076259797983873"]'}], 'negative_passages': [{'text': '杭州中新街', 'gis': '[[], [], [], [], [], "120.20509044775532,30.076259797983873"]'}]}使用的是网站的例子

展开
收起
真的很搞笑 2023-10-08 18:46:56 169 分享 版权
1 条回答
写回答
取消 提交回答
  • 根据您提供的信息,报错信息显示在执行自定义训练时出现了两个错误。

    第一个错误是关于张量尺寸不匹配的问题。根据错误信息,期望的张量尺寸是20,但实际得到的尺寸是2。这可能是由于数据加载或预处理过程中出现了问题。您可以检查train.json和dev.json文件中的数据是否正确,并确保它们与模型要求的输入格式相匹配。

    第二个错误是关于指标注册的问题。错误信息指出'mrr@1'指标未在指标注册表组中注册。这可能是因为您使用的ModelScope版本与示例代码不兼容导致的。请确保您使用的是正确版本的ModelScope库,并且指标名称是正确的。

    针对这两个错误,您可以尝试以下解决方法:

    1. 检查train.json和dev.json文件中的数据,确保其格式正确且与模型要求的输入一致。
    2. 确保您使用的ModelScope版本与示例代码兼容,并且指标名称正确。
    2023-10-10 12:47:32
    赞同 展开评论

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

热门讨论

热门文章

还有其他疑问?
咨询AI助理