生成模型

  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