pycham debug accelerate的方法(accelerate launch)
例如下面这个命令怎么运行?可以进行debug什么的
CUDA_VISIBLE_DEVICES=0 accelerate launch --config_file 8_gpu.json --main_process_port 25655 train_face.py \
--pretrained_model_name_or_path="runwayml/stable-diffusion-v1-5" \
--data_json_file=None \
--data_root_path=None \
--mixed_precision="fp16" \
--resolution=512 \
--train_batch_size=16 \
--dataloader_num_workers=4 \
--learning_rate=1e-04 \
--weight_decay=0.01 \
--output_dir="./experiments_stage1" \
--save_steps=50 \
方法:
选择解释器,此处为MDT解释器
选择为module而不是普通的script
输入 accelerate.commands.launch ,这个不能变成其他的
输入其他配置:--XX --YY
CUDA_VISIBLE_DEVICES=0 直接输入进environment variables中