From 0cfa39a37359e500cbaa699ec15ebfe8f411238f Mon Sep 17 00:00:00 2001 From: Frederik Fix Date: Thu, 15 Sep 2022 19:46:52 +0200 Subject: [PATCH] Update train.sh --- train.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.sh b/train.sh index b35545b..13144e9 100644 --- a/train.sh +++ b/train.sh @@ -1 +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 +python3 main.py --train --resume model.ckpt --base ./configs/stable-diffusion/v1-finetune-4gpu.yaml --no-test --seed 25 --scale_lr False --gpus 0,1,2,3