| export MODEL_DIR="../modelsaved/Tiger-Corase/checkpoint-xxx" |
| export OUTPUT_DIR="/modelsaved/Tiger-Fine" |
| export DATASET="../dataset/training data/init_image" |
|
|
| CUDA_VISIBLE_DEVICES='0,1,2' accelerate launch --multi_gpu --num_processes=3 Tiger Model/Fine-Training.py \ |
| --pretrained_model_name_or_path=$MODEL_DIR \ |
| --output_dir=$OUTPUT_DIR \ |
| --dataset_name=$DATASET \ |
| --resolution=512 \ |
| --seed=2023 \ |
| --learning_rate=1e-5 \ |
| --gradient_accumulation_steps=4 \ |
| --checkpointing_steps=200 \ |
| --max_train_steps=100000000000 \ |
| --train_batch_size=24 \ |
| --image_column="image" \ |
| --caption_column_nd="text_nd" --caption_column_bg="text_bd" \ |
| --conditioning_nd_column="condition_FG" --conditioning_bg_column="condition_BG" \ |
| --validation_image "../valid_Figure.png" \ |
| --validation_prompt "malignant papillary solid ..." \ |
| --validation_steps=500 |
|
|
