File size: 862 Bytes
9627ce0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | SUBJECT_TEXT="dog"
IMAGE_STORAGE="/export/home/workspace/LAVIS-Diffusion/LAVIS/projects/blip-diffusion/images/dreambooth/dog"
MAX_ITERS=40
ITERS_PER_INNER_EPOCH=40 # number of iterations before saving a checkpoint
BATCH_SIZE=3
LR=5e-6
WEIGHT_DECAY=0.01
OUTPUT_DIR="output/debug/BLIP-diffusion/finetune/dog"
python -m torch.distributed.run \
--nproc_per_node=1 train.py \
--cfg-path lavis/projects/blip_diffusion/finetune-db-template.yaml \
--options datasets.blip_diffusion_finetune.build_info.subject_text=$SUBJECT_TEXT \
datasets.blip_diffusion_finetune.build_info.images.storage=$IMAGE_STORAGE \
run.max_iters=$MAX_ITERS \
run.iters_per_inner_epoch=$ITERS_PER_INNER_EPOCH \
run.output_dir=$OUTPUT_DIR \
run.init_lr=$LR \
run.weight_decay=$WEIGHT_DECAY \
run.batch_size_train=$BATCH_SIZE
|