华为NPU单卡训练命令

!RANK=0 WORLD_SIZE=1 MASTER_ADDR=127.0.0.1 MASTER_PORT=任意端口号\
python 执行脚本 --model 模型名称 --data-path 指定训练数据的路径 --vae 变分自编码器类型\
--global-batch-size 1 --epochs 1 --global-seed 0 --log-every 1

参数说明：

• --model ：指定要训练的模型
• --data-path 指定训练数据的路径
• --vae：表明使用 VAE（变分自编码器）
• --global-batch-size 1：设置全局批次大小为1，表示每次训练只使用1个样本进行训练，适合于资源有限的情况。
• --epochs 1：指定训练的轮数为1，表示整个数据集将被遍历一次。
• --global-seed 0：设置全局随机种子为0，确保实验的可重复性，即每次运行时结果相同。
• --log-every 1：每1个训练周期记录日志一次，这通常用于跟踪训练过程中的性能。