From df8761695bedc63167a8e4af8573c7cdf4a7029f Mon Sep 17 00:00:00 2001 From: harubaru Date: Thu, 8 Sep 2022 18:46:23 -0700 Subject: [PATCH] redo training launcher --- train.bat | 1 - train.sh | 1 + 2 files changed, 1 insertion(+), 1 deletion(-) delete mode 100644 train.bat create mode 100644 train.sh diff --git a/train.bat b/train.bat deleted file mode 100644 index b52c776..0000000 --- a/train.bat +++ /dev/null @@ -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" \ No newline at end of file diff --git a/train.sh b/train.sh new file mode 100644 index 0000000..b35545b --- /dev/null +++ b/train.sh @@ -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