昇腾社区首页
中文
注册

生成模型

  1. 创建BuilderConfig。
    builder_config = aie.BuilderConfig()
  2. 配置BuilderConfig类设置编译选项。
    builder_config.flag = aie.BuilderFlag.fp16
  3. 调用builder的build_model接口完成对模型的生成。
    data = builder.build_model(network, builder_config)
  4. (可选)保存生成的离线模型。
    with open('resnet50.om', 'wb') as f:
        f.write(data.data)
  5. 销毁资源。
    del builder