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
}