| 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 | |