File size: 909 Bytes
092fe0d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
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
|