File size: 1,025 Bytes
b084524 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
{
"project_dir": "./exp_output/test_baseline_style_omni150k_1",
"mixed_precision": "bf16",
"gradient_accumulation_steps": 1,
"seed": 42,
"wandb_project_name": null,
"wandb_run_name": null,
"model_name": "flux-dev",
"lora_rank": 128,
"double_blocks_indices": null,
"single_blocks_indices": null,
"pe": "d",
"gradient_checkpoint": false,
"ema": false,
"ema_interval": 1,
"ema_decay": 0.99,
"learning_rate": 0.0001,
"adam_betas": [
0.9,
0.999
],
"adam_eps": 1e-08,
"adam_weight_decay": 0.01,
"max_grad_norm": 1.0,
"lr_scheduler": "constant",
"lr_warmup_steps": 50,
"max_train_steps": 10000,
"train_data_json": "datasets/omni150k.json",
"batch_size": 2,
"text_dropout": 0.1,
"resolution": 512,
"resolution_ref": 224,
"eval_data_json": "datasets/styleeval0802.json",
"eval_batch_size": 1,
"resume_from_checkpoint": "latest",
"checkpointing_steps": 1000,
"log_image_steps": 100
} |