Skip to content

Commit

Permalink
low batchsize
Browse files Browse the repository at this point in the history
  • Loading branch information
caradryanl committed May 23, 2024
1 parent 72f05d8 commit a21c9a1
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions diffusers/scripts/exp_ldm_sd_eval.sh
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
python scripts/train_secmi.py --model-type sd --ckpt-path ../models/diffusers/stable-diffusion-v1-5/ --member-dataset laion-aesthetic-2-5k-eval --holdout-dataset coco2017-val-2-5k-eval --eval True --batch-size 24
python scripts/train_pia.py --model-type sd --ckpt-path ../models/diffusers/stable-diffusion-v1-5/ --member-dataset laion-aesthetic-2-5k-eval --holdout-dataset coco2017-val-2-5k-eval --eval True --batch-size 24
python scripts/train_pfami.py --model-type sd --ckpt-path ../models/diffusers/stable-diffusion-v1-5/ --member-dataset laion-aesthetic-2-5k-eval --holdout-dataset coco2017-val-2-5k-eval --eval True --batch-size 24
python scripts/train_secmi.py --model-type ldm --ckpt-path ../models/diffusers/ldm-celebahq-256/ --member-dataset celeba-hq-2-5k-eval --holdout-dataset ffhq-2-5k-eval --eval True --batch-size 32
python scripts/train_pia.py --model-type ldm --ckpt-path ../models/diffusers/ldm-celebahq-256/ --member-dataset celeba-hq-2-5k-eval --holdout-dataset ffhq-2-5k-eval --eval True --batch-size 32
python scripts/train_pfami.py --model-type ldm --ckpt-path ../models/diffusers/ldm-celebahq-256/ --member-dataset celeba-hq-2-5k-eval --holdout-dataset ffhq-2-5k-eval --eval True --batch-size 32
python scripts/train_secmi.py --model-type sd --ckpt-path ../models/diffusers/stable-diffusion-v1-5/ --member-dataset laion-aesthetic-2-5k-eval --holdout-dataset coco2017-val-2-5k-eval --eval True --batch-size 64
python scripts/train_pia.py --model-type sd --ckpt-path ../models/diffusers/stable-diffusion-v1-5/ --member-dataset laion-aesthetic-2-5k-eval --holdout-dataset coco2017-val-2-5k-eval --eval True --batch-size 64
python scripts/train_pfami.py --model-type sd --ckpt-path ../models/diffusers/stable-diffusion-v1-5/ --member-dataset laion-aesthetic-2-5k-eval --holdout-dataset coco2017-val-2-5k-eval --eval True --batch-size 64
python scripts/train_secmi.py --model-type ldm --ckpt-path ../models/diffusers/ldm-celebahq-256/ --member-dataset celeba-hq-2-5k-eval --holdout-dataset ffhq-2-5k-eval --eval True --batch-size 96
python scripts/train_pia.py --model-type ldm --ckpt-path ../models/diffusers/ldm-celebahq-256/ --member-dataset celeba-hq-2-5k-eval --holdout-dataset ffhq-2-5k-eval --eval True --batch-size 96
python scripts/train_pfami.py --model-type ldm --ckpt-path ../models/diffusers/ldm-celebahq-256/ --member-dataset celeba-hq-2-5k-eval --holdout-dataset ffhq-2-5k-eval --eval True --batch-size 96

0 comments on commit a21c9a1

Please sign in to comment.