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

请问怎么调用blade.optimize函数来模型优化?

已解决

请问怎么调用blade.optimize函数来模型优化?

展开
收起
felix@ 2023-01-29 12:41:49 1023 1
1 条回答
写回答
取消 提交回答
  • 今天也要加油吖~
    采纳回答

    您好,模型优化代码可以参照:

    optimized_model, opt_spec, report = blade.optimize(
        model,                 # 待优化的模型。
        'o1',                  # 优化级别,o1或o2。
        device_type='gpu',     # 目标设备,gpu/cpu。
        test_data=[(dummy,)],  # PyTorch的输入数据是List of tuple of tensor。
    )
    
    2023-01-29 12:46:10
    赞同 展开评论 打赏

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

相关电子书

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