Update train.sh

2022-09-15 19:46:52 +02:00 · 2022-09-15 19:46:52 +02:00 · 0cfa39a373
parent ab67872afa
commit 0cfa39a373
1 changed files with 1 additions and 1 deletions
--- 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