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"]'}]}使用的是网站的例子
根据您提供的信息,报错信息显示在执行自定义训练时出现了两个错误。
第一个错误是关于张量尺寸不匹配的问题。根据错误信息,期望的张量尺寸是20,但实际得到的尺寸是2。这可能是由于数据加载或预处理过程中出现了问题。您可以检查train.json和dev.json文件中的数据是否正确,并确保它们与模型要求的输入格式相匹配。
第二个错误是关于指标注册的问题。错误信息指出'mrr@1'指标未在指标注册表组中注册。这可能是因为您使用的ModelScope版本与示例代码不兼容导致的。请确保您使用的是正确版本的ModelScope库,并且指标名称是正确的。
针对这两个错误,您可以尝试以下解决方法: