Table 2 LoRA training parameter configuration

From: Diffusion model-based image generation method for Cantonese embroidery artistic styles

Item

Parameter

Dataset

494 high-quality images of Cantonese embroidery works

Semantic Types

kapok flowers, plum blossoms, chrysanthemums, orchids, lotus flowers, peach blossoms, roses, peonies, leaves, chicks, roosters, hens, peacocks, pheasants, mandarin ducks, magpies, sparrows, rhododendrons, goldfish, branches, lychees, peaches, pomegranates, grapes, humans, horses, pandas, tigers, dogs, sheep, squirrels, monkeys, butterflies, bamboo, banana plants, willows, maple trees, pine trees, pumpkins, palaces, gardens, temples, mountains, water, beaches

Model_train_type

sd-lora

Pre-trained Model

v1-5-pruned-emaonly.safetensors

Enable_bucket

Min _bucket_reso:512

 

Max _bucket_reso:1024

 

Bucket_reso_steps:64

Resolution

512,512

Model output

gx_lora3.safetensors

Save_precision

fp16

Rounds

12

Max Train Epochs

10

Train _batch_size

2

Learning_rate

Global:1e-4

 

U-Net lr:1e-4

 

Text Encoder:1e-5

 

Scheduler:cosine_with_restarts

Optimizer_type

Adamw8bit

Network

Module:networks.lora

 

Network_dim:64

 

Network_alpha:64

 

Network_dropout:0