redo training launcher

This commit is contained in:
harubaru 2022-09-08 18:46:23 -07:00 committed by GitHub
parent 952bdc04df
commit df8761695b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 1 additions and 1 deletions

View File

@ -1 +0,0 @@
python main.py --base "configs/stable-diffusion/v1-finetune.yaml" -t --no-test --actual-resume "./models/ldm/stable-diffusion/model.ckpt" -n "waifu-diffusion" --gpus 1 --data_root "./danbooru-aesthetic"

1
train.sh Normal file
View File

@ -0,0 +1 @@
python3 main.py --resume model.ckpt --base ./configs/stable-diffusion/v1-finetune-4gpu.yaml --no-test --seed 25 --scale_lr False --gpus 0,1,2,3