AI开发平台MODELARTS-SDXL基于DevServer适配PyTorch NPU的Finetune训练指导(6.3.905):Step5 修改算法脚本

时间:2024-12-09 20:36:09

Step5 修改算法脚本

进入容器后,修改启动脚本文件。

vi /home/ma-user/sdxl-train/user-job-dir/code/diffusers_finetune_train.sh

在第2行增加export MA_NUM_HOSTS=1 即可,如:

#!/bin/bash
export MA_NUM_HOSTS=1
if [[ $MA_NUM_HOSTS == 1 ]]; then
support.huaweicloud.com/bestpractice-modelarts/modelarts_10_01201.html