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