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

ModelScope中,dummy_inputs有哪些格式?

ModelScope中,dummy_inputs有哪些格式? 我输入字典,存在错误。 这里面格式转换有点多了lQLPJxUwEwmAnDjM0M0D8bB1qWzgAR8_GQUWgfmHQLYA_1009_208.png

展开
收起
多麻辣哦 2023-10-18 20:22:12 92 0
2 条回答
写回答
取消 提交回答
  • 在ModelScope中,dummy_inputs可以有多种格式。以下是一些常见的dummy_inputs格式示例:

    1. 单个输入的格式:

      dummy_inputs = {
          'input_name': input_value
      }
      
    2. 多个输入的格式:

      dummy_inputs = {
          'input_name1': input_value1,
          'input_name2': input_value2,
          ...
      }
      
    3. 对于包含多个输入的模型,可以使用列表或元组来表示多个输入的不同取值组合:

      dummy_inputs = [
          {'input_name1': input_value1, 'input_name2': input_value2},
          {'input_name1': input_value3, 'input_name2': input_value4},
          ...
      ]
      
    4. 对于多个输入的模型,还可以使用字典列表来表示每个输入取值的多个组合:

      dummy_inputs = {
          'input_name1': [input_value1, input_value2, ...],
          'input_name2': [input_value3, input_value4, ...],
          ...
      }
      

    请根据实际模型的输入情况选择合适的dummy_inputs格式。在为模型生成dummy_inputs时,需要根据模型的输入参数和输入数据类型进行相应的填写。

    2023-10-19 14:53:30
    赞同 展开评论 打赏
  • from modelscope import AutoModelForSequenceClassification, AutoTokenizer
    model = AutoModelForSequenceClassification.from_pretrained('AI-ModelScope/bert-base-uncased')
    tokenizer = AutoTokenizer.from_pretrained('AI-ModelScope/bert-base-uncased')
    from modelscope.exporters import TorchModelExporter
    from collections import OrderedDict

    假设最大支持256长度的句子

    dummy_inputs = tokenizer(tokenizer.unk_token, padding='max_length', max_length=256, return_tensors='pt')
    dynamic_axis = {0: 'batch', 1: 'sequence'}
    inputs = OrderedDict([
    ('input_ids', dynamic_axis),
    ('attention_mask', dynamic_axis),
    ('token_type_ids', dynamic_axis),
    ])
    outputs = OrderedDict({'logits': {0: 'batch'}})
    output_files = TorchModelExporter().export_onnx(model=model, dummy_inputs=dummy_inputs, inputs=inputs,
    outputs=outputs, output_dir='/tmp')
    print(output_files) # {'model': '/tmp/model.onnx'}这是文档中的例子,传入的就是一个dict——此回答整理自钉群:魔搭ModelScope开发者联盟群 ①

    2023-10-18 21:43:09
    赞同 展开评论 打赏

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

热门讨论

热门文章

相关电子书

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