search

XLabs-AI / x-flux

Apache License 2.0
1.65k stars 118 forks source link

dataset class and configuration for open pose training #118

Open raulc0399 opened 2 months ago

raulc0399 commented 2 months ago

create a yaml file like to start the open pose training

model_name: "flux-dev" is_openpose: true data_config: train_batch_size: 2 num_workers: 2 img_size: 512 img_dir: images/ report_to: wandb train_batch_size: 2 output_dir: saves_openpose/ max_train_steps: 100000 learning_rate: 2e-5 lr_scheduler: constant lr_warmup_steps: 10 adam_beta1: 0.9 adam_beta2: 0.999 adam_weight_decay: 0.01 adam_epsilon: 1e-8 max_grad_norm: 1.0 logging_dir: logs mixed_precision: "bf16" checkpointing_steps: 2500 checkpoints_total_limit: 50 tracker_project_name: openpose_training resume_from_checkpoint: latest gradient_accumulation_steps: 2

tianqyun111 commented 1 week ago

i trained pose controlnet with 50000 images,but after trained 2epoch, when inference, even i set strength to 1,The image does not have any guided effect.Anyone can help me?