File size: 393 Bytes
d807efd
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
export IMAGE_NAME="example1"
export IMAGE_NAME_2="example2"
python main.py --name=$IMAGE_NAME \
    --dpm="sd" \
    --resolution=512 \
    --image \
    --name_2=$IMAGE_NAME_2 \
    --embedding_learning_rate=1e-4 \
    --diffusion_model_learning_rate=5e-5 \
    --max_emb_train_steps=500 \
    --max_diffusion_train_steps=500 \
    --train_batch_size=5 \
    --gradient_accumulation_steps=5