File size: 592 Bytes
4714609 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
args = Namespace(
image_path="/content/images/CFP",
output_path="/content/output",
prior_model="kandinsky-community/kandinsky-2-2-prior",
decoder_model="kandinsky-community/kandinsky-2-2-decoder",
gradient_accumulation_steps=1,
mixed_precision='bf16',
seed=1703,
size=256,
rank=32,
use_rslora=True,
init_lora_weights='gaussian',
max_grad_norm=1,
lr=1e-4,
lr_scheduler='constant',
lr_warmup_steps=0,
train_batch_size=32,
num_train_epochs=10,
snr_gamma=5,
resume_from_checkpoint=None,
checkpointing_steps=680
) |