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

白盒化加速设置加速开关等

已解决

请问白盒化加速如何设置--user-script为rapidformer_finetune_huggingface_bert_trainer.py,并设置加速开关。

展开
收起
felix@ 2023-01-28 20:22:23 588 0
2 条回答
写回答
取消 提交回答
  • 今天也要加油吖~
    采纳回答

    您好,可以按如下操作设置:

    #!/bin/bash
    export CUDA_VISIBLE_DEVICES=4,5,6,7
    export MASTER_ADDR=localhost
    export MASTER_PORT=6010
    export NNODES=1
    export NODE_RANK=0
    
    rapidformer --user-script rapidformer_finetune_huggingface_bert_trainer.py
                --task sequence_classification \
                --pretrained-model-name-or-path 'bert-base-cased' \
                --data-path glue \
                --data-name mrpc \
                --epochs 3 \
                --micro-batch-size 16 \
                --global-batch-size 16 \
                --lr 2e-5 \
                --lr-decay-style linear \
                --lr-warmup-iters 100 \
                --weight-decay 1e-2 \
                --clip-grad 1.0 \
                --mixed-precision                                 #开启混合精度训练
                --zero-3-memory-optimization \                    #开启模型状态切分
                --onnx-runtime-training \                         #开启计算图优化
    
    2023-01-28 20:29:26
    赞同 展开评论 打赏
  • 白盒化加速设置步骤如下: 1. 在命令行中输入以下命令:

    python -m rapidformer_finetune_huggingface_bert_trainer --user-script rapidformer_finetune_huggingface_bert_trainer.py --acceleration-switch on
    
    1. 在训练过程中,可以通过查看日志文件来确认白盒化加速是否已经开启。
    2. 如果白盒化加速已经开启,可以在训练过程中观察到模型训练速度的提升。
    2023-01-28 21:21:17
    赞同 展开评论 打赏

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

热门讨论

热门文章

相关电子书

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