开发者社区 > 大数据与机器学习 > 人工智能平台PAI > 正文

请问用Blade优化RetinaNet时怎么调用blade.optimize接口对模型进行优化?

已解决

请问用Blade优化RetinaNet时怎么调用blade.optimize接口对模型进行优化?

展开
收起
felix@ 2023-01-29 13:01:51 701 2
2 条回答
写回答
取消 提交回答
  • 今天也要加油吖~
    采纳回答

    您好,代码可以参照:

    import blade
    
    test_data = [(batched_inputs,)] # PyTorch的输入数据是List of tuple。
    optimized_model, opt_spec, report = blade.optimize(
        script_model,  # 上一步导出的TorchScript模型。 
        'o1',  # 开启Blade O1级别的优化。
        device_type='gpu', # 目标设备为GPU。
        test_data=test_data, # 给定一组测试数据,用于辅助优化及测试。
    )
    
    2023-01-29 13:05:27
    赞同 1 展开评论 打赏
  • 我为你总结操作流程如下:

    创建Blade配置文件,指定模型文件和数据配置。

    使用blade.optimize接口对模型进行优化,可以设置优化的目标(如减少模型大小、提高推理速度等)和优化策略。

    运行优化过程,并导出优化后的模型。

    使用优化后的模型进行推理,并验证其性能。

    2023-01-29 16:12:08
    赞同 展开评论 打赏

人工智能平台 PAI(Platform for AI,原机器学习平台PAI)是面向开发者和企业的机器学习/深度学习工程平台,提供包含数据标注、模型构建、模型训练、模型部署、推理优化在内的AI开发全链路服务,内置140+种优化算法,具备丰富的行业场景插件,为用户提供低门槛、高性能的云原生AI工程化能力。

热门讨论

热门文章

相关电子书

更多
低代码开发师(初级)实战教程 立即下载
冬季实战营第三期:MySQL数据库进阶实战 立即下载
阿里巴巴DevOps 最佳实践手册 立即下载