From 13c8278ee6361789c5c22d7f6a18baa270e8338e Mon Sep 17 00:00:00 2001 From: CaradryanLiang <caradryan2022@gmail.com> Date: Thu, 16 May 2024 18:08:39 -0700 Subject: [PATCH] smaller bsz for gsa --- diffusers/scripts/exp_ldm_sd_gsa.sh | 8 ++++---- diffusers/scripts/exp_ldm_sd_gsa_eval.sh | 8 ++++---- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/diffusers/scripts/exp_ldm_sd_gsa.sh b/diffusers/scripts/exp_ldm_sd_gsa.sh index 81ea134..ed69b37 100755 --- a/diffusers/scripts/exp_ldm_sd_gsa.sh +++ b/diffusers/scripts/exp_ldm_sd_gsa.sh @@ -1,4 +1,4 @@ -python scripts/train_gsa.py --model-type sd --gsa-mode 1 --ckpt-path ../models/diffusers/stable-diffusion-v1-5/ --member-dataset laion-aesthetic-2-5k --holdout-dataset coco2017-val-2-5k --batch-size 5 -python scripts/train_gsa.py --model-type sd --gsa-mode 2 --ckpt-path ../models/diffusers/stable-diffusion-v1-5/ --member-dataset laion-aesthetic-2-5k --holdout-dataset coco2017-val-2-5k --batch-size 5 -python scripts/train_gsa.py --model-type ldm --gsa-mode 1 --ckpt-path ../models/diffusers/ldm-celebahq-256/ --member-dataset celeba-hq-2-5k --holdout-dataset ffhq-2-5k --batch-size 5 -python scripts/train_gsa.py --model-type ldm --gsa-mode 2 --ckpt-path ../models/diffusers/ldm-celebahq-256/ --member-dataset celeba-hq-2-5k --holdout-dataset ffhq-2-5k --batch-size 5 \ No newline at end of file +python scripts/train_gsa.py --model-type sd --gsa-mode 1 --ckpt-path ../models/diffusers/stable-diffusion-v1-5/ --member-dataset laion-aesthetic-2-5k --holdout-dataset coco2017-val-2-5k --batch-size 2 +python scripts/train_gsa.py --model-type sd --gsa-mode 2 --ckpt-path ../models/diffusers/stable-diffusion-v1-5/ --member-dataset laion-aesthetic-2-5k --holdout-dataset coco2017-val-2-5k --batch-size 2 +python scripts/train_gsa.py --model-type ldm --gsa-mode 1 --ckpt-path ../models/diffusers/ldm-celebahq-256/ --member-dataset celeba-hq-2-5k --holdout-dataset ffhq-2-5k --batch-size 2 +python scripts/train_gsa.py --model-type ldm --gsa-mode 2 --ckpt-path ../models/diffusers/ldm-celebahq-256/ --member-dataset celeba-hq-2-5k --holdout-dataset ffhq-2-5k --batch-size 2 \ No newline at end of file diff --git a/diffusers/scripts/exp_ldm_sd_gsa_eval.sh b/diffusers/scripts/exp_ldm_sd_gsa_eval.sh index 08ba11d..8675dad 100755 --- a/diffusers/scripts/exp_ldm_sd_gsa_eval.sh +++ b/diffusers/scripts/exp_ldm_sd_gsa_eval.sh @@ -1,4 +1,4 @@ -python scripts/train_gsa.py --model-type sd --gsa-mode 1 --ckpt-path ../models/diffusers/stable-diffusion-v1-5/ --member-dataset laion-aesthetic-2-5k --holdout-dataset coco2017-val-2-5k --eval True --batch-size 5 -python scripts/train_gsa.py --model-type sd --gsa-mode 2 --ckpt-path ../models/diffusers/stable-diffusion-v1-5/ --member-dataset laion-aesthetic-2-5k --holdout-dataset coco2017-val-2-5k --eval True --batch-size 5 -python scripts/train_gsa.py --model-type ldm --gsa-mode 1 --ckpt-path ../models/diffusers/ldm-celebahq-256/ --member-dataset celeba-hq-2-5k --holdout-dataset ffhq-2-5k --eval True --batch-size 5 -python scripts/train_gsa.py --model-type ldm --gsa-mode 2 --ckpt-path ../models/diffusers/ldm-celebahq-256/ --member-dataset celeba-hq-2-5k --holdout-dataset ffhq-2-5k --eval True --batch-size 5 \ No newline at end of file +python scripts/train_gsa.py --model-type sd --gsa-mode 1 --ckpt-path ../models/diffusers/stable-diffusion-v1-5/ --member-dataset laion-aesthetic-2-5k --holdout-dataset coco2017-val-2-5k --eval True --batch-size 2 +python scripts/train_gsa.py --model-type sd --gsa-mode 2 --ckpt-path ../models/diffusers/stable-diffusion-v1-5/ --member-dataset laion-aesthetic-2-5k --holdout-dataset coco2017-val-2-5k --eval True --batch-size 2 +python scripts/train_gsa.py --model-type ldm --gsa-mode 1 --ckpt-path ../models/diffusers/ldm-celebahq-256/ --member-dataset celeba-hq-2-5k --holdout-dataset ffhq-2-5k --eval True --batch-size 2 +python scripts/train_gsa.py --model-type ldm --gsa-mode 2 --ckpt-path ../models/diffusers/ldm-celebahq-256/ --member-dataset celeba-hq-2-5k --holdout-dataset ffhq-2-5k --eval True --batch-size 2 \ No newline at end of file