github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

modify shell for check

pull/3905/head
Maruyama_Aya 2023-06-08 13:17:58 +08:00
parent cf4792c975
commit 039854b391
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
examples/images/dreambooth/test_ci.sh
View File

@ -6,8 +6,8 @@ HF_DATASETS_OFFLINE=1
TRANSFORMERS_OFFLINE=1 TRANSFORMERS_OFFLINE=1
DIFFUSERS_OFFLINE=1 DIFFUSERS_OFFLINE=1
# "torch_ddp" "torch_ddp_fp16" # "torch_ddp" "torch_ddp_fp16" "low_level_zero"
for plugin in "low_level_zero" "gemini"; do for plugin in "gemini"; do
torchrun --nproc_per_node 4 --standalone train_dreambooth_colossalai.py \ torchrun --nproc_per_node 4 --standalone train_dreambooth_colossalai.py \
--pretrained_model_name_or_path="/data/dreambooth/diffuser/stable-diffusion-v1-4" \ --pretrained_model_name_or_path="/data/dreambooth/diffuser/stable-diffusion-v1-4" \
--instance_data_dir="/data/dreambooth/Teyvat/data" \ --instance_data_dir="/data/dreambooth/Teyvat/data" \
@ -20,5 +20,5 @@ for plugin in "low_level_zero" "gemini"; do
--lr_scheduler="constant" \ --lr_scheduler="constant" \
--lr_warmup_steps=0 \ --lr_warmup_steps=0 \
--num_class_images=200 \ --num_class_images=200 \
--placement="cuda" --placement="cpu" # "cuda"
done done

2
examples/images/dreambooth/train_dreambooth_colossalai.py
View File

@ -487,7 +487,7 @@ def main(args):
if args.plugin.startswith('torch_ddp'): if args.plugin.startswith('torch_ddp'):
plugin = TorchDDPPlugin() plugin = TorchDDPPlugin()
elif args.plugin == 'gemini': elif args.plugin == 'gemini':
plugin = GeminiPlugin(placement_policy='cuda', strict_ddp_mode=True, initial_scale=2 ** 5) plugin = GeminiPlugin(placement_policy=args.placement, strict_ddp_mode=True, initial_scale=2 ** 5)
elif args.plugin == 'low_level_zero': elif args.plugin == 'low_level_zero':
plugin = LowLevelZeroPlugin(initial_scale=2 ** 5) plugin = LowLevelZeroPlugin(initial_scale=2 ** 5)