From a1ef156dba2197862d6ab8f9ca95f99ffbf7bf73 Mon Sep 17 00:00:00 2001 From: CaradryanLiang Date: Mon, 13 May 2024 14:58:44 -0700 Subject: [PATCH] correct ldm setups --- diffusers/scripts/exp_ldm.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/diffusers/scripts/exp_ldm.sh b/diffusers/scripts/exp_ldm.sh index 8dff0f8..1a8d81b 100755 --- a/diffusers/scripts/exp_ldm.sh +++ b/diffusers/scripts/exp_ldm.sh @@ -1,2 +1,2 @@ -python scripts/train_secmi.py --model-type ldm --ckpt-path ../models/diffusers/ldm-celebahq-256/ --member-dataset celeba-hq-2-5k --holdout-dataset ffhq-2-5k -python scripts/train_pia.py --model-type ldm --ckpt-path ../models/diffusers/ldm-celebahq-256/ --member-dataset celeba-hq-2-5k --holdout-dataset ffhq-2-5k \ No newline at end of file +python scripts/train_secmi.py --model-type ldm --ckpt-path ../models/diffusers/ldm-celebahq-256/ --member-dataset celeba-hq-2-5k --holdout-dataset ffhq-2-5k --batch-size 32 +python scripts/train_pia.py --model-type ldm --ckpt-path ../models/diffusers/ldm-celebahq-256/ --member-dataset celeba-hq-2-5k --holdout-dataset ffhq-2-5k --batch-size 32 \ No newline at end of file