AI开发平台MODELARTS-SDXL基于Standard适配PyTorch NPU的LoRA训练指导（6.3.907）:Step7 创建训练作业

时间：2024-09-14 22:29:38

AI开发平台MODELARTS

创建训练作业，填下如下参数。

创建方式：选择自定义算法，启动方式选择自定义，然后选择上传到SWR的自定义镜像。
代码目录：选择上传到OBS的代码文件夹，例如/sdxl-train/code。如果用户需要修改代码文件，可修改OBS桶中代码文件，创建训练作业时，会将OBS的code目录复制到训练容器的/home/ma-user/sdxl-train/user-job-dir/目录下，覆盖容器中原有的code目录。

启动命令：将华为侧优化后代码文件复制到工作目录后，运行启动脚本文件diffusers_sdxl_lora_train.sh。

cd /home/ma-user/sdxl-train/user-job-dir/code && cp /home/ma-user/sdxl-train/train_text_to_image_lora_sdxl.py ./ && sh diffusers_sdxl_lora_train.sh

本地代码目录：保持默认即可。
工作目录：选择代码文件目录，例如/home/ma-user/sdxl-train/user-job-dir/code/。
输出：单击“增加训练输出”，将模型保存到OBS中。参数名称为output，数据存储位置选择OBS桶中制定文件夹，例如sdxl-train/checkpoint，获取方式选择环境变量，/home/ma-user/modelarts/outputs/output_0下的模型文件会保存到OBS中。
图4 选择镜像