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