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

ModelScope的souce_sentence的list大小有要求吗,可以塞多少个性能没问题?

ModelScope的souce_sentence的list大小有要求吗,可以塞多少个性能没问题?

展开
收起
青城山下庄文杰 2023-08-12 10:24:08 37 0
2 条回答
写回答
取消 提交回答
  • 看你的显卡显存大小,32G的V100 128长度建议list大小128-此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”

    2023-08-15 08:22:55
    赞同 展开评论 打赏
  • 北京阿里云ACE会长

    ModelScope的source_sentence接受一个列表作为输入,列表大小默认没有明确限制,但实际上会影响性能和内存占用:

    列表越长,传输和处理数据量就越大,性能和响应时间会下降。

    一般来说,个别模型有自己的batch大小限制,超过这个限制可能会报错。

    ModelScope内部会将列表转化为批处理输入模型,所以列表长度会直接影响批大小。

    太长的列表可能会触发内存限制导致模型无法处理。

    一般来说:

    BERT系列和GPT系列模型,源列表长度最大在100个左右可以保证性能。

    Transformer模型可能支持稍长一些,到几百个没问题。

    极端情况下若只有几个字符串,列表长度10000多也未必有明显影响。

    所以具体情况下:

    推荐源列表长度控制在100个以下。

    可以适当增加,但注意响应时间和内存占用。

    如果集群有明确限制,不超过那个限制。

    如遇内存报错,可以考虑缩小批大小解决。

    2023-08-13 16:43:36
    赞同 展开评论 打赏

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

相关电子书

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