| { | |
| "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 | |
| } |