SMPL_sd15_config.json

{
  "config_save_dir": "output/configs",
  "mixed_precision_type": "no",
  "num_processes": 12,
  "num_machines": 1,
  "num_cpu_threads_per_process": 4,
  "dynamo_backend": "no",
  "multi_gpu": false,
  "gpu_ids": "",
  "main_process_port": 0,
  "extra_args": "",
  "pretrained_model_name_or_path": "yeq6x/v1-5-pruned_pls_SMPL_LoRA_sd1.5_v1.fp16",
  "pretrained_vae_model_name_or_path": "",
  "controlnet_model_name_or_path": "",
  "variant": "",
  "revision": "",
  "tokenizer_name": "",
  "output_dir": "/workspace/cn_output/2025-02-19-14-08",
  "cache_dir": "/workspace/cn_output/cache",
  "dataset_name": "yeq6x/SynthMoCap_photo_to_smpl_512",
  "train_data_dir": "",
  "dataset_config_name": "",
  "dataloader_num_workers": 0,
  "image_column": "image",
  "conditioning_image_column": "conditioning_image",
  "caption_column": "text",
  "max_train_samples": 50000,
  "proportion_empty_prompts": 0,
  "train_batch_size": 32,
  "max_train_steps": 50000,
  "num_train_epochs": null,
  "seed": 43,
  "resolution": 512,
  "crops_coords_top_left_h": 0,
  "crops_coords_top_left_w": 0,
  "learning_rate": 0.0001,
  "scale_lr": false,
  "lr_scheduler": "constant",
  "lr_warmup_steps": null,
  "lr_num_cycles": 1,
  "lr_power": 1,
  "gradient_accumulation_steps": 1,
  "gradient_checkpointing": false,
  "use_8bit_adam": false,
  "adam_beta1": 0.9,
  "adam_beta2": 0.999,
  "adam_weight_decay": 0.01,
  "adam_epsilon": 1e-08,
  "max_grad_norm": 1,
  "set_grads_to_none": false,
  "checkpointing_steps": 5000,
  "checkpoints_total_limit": null,
  "resume_from_checkpoint": "",
  "num_validation_images": 2,
  "validation_steps": 1000,
  "validation_prompt": [
    "yellow smpl, hand, black background",
    "blue smpl, hand, black background",
    "green smpl, hand, black background",
    "orange smpl, full body, black background",
    "yellow smpl, full body, black background",
    "yellow smpl, full body, black background"
  ],
  "validation_image": [
    "/workspace/test_img/image_000000.png",
    "/workspace/test_img/image_000007.png",
    "/workspace/test_img/image_000030.png",
    "/workspace/test_img/image_030265.png",
    "/workspace/test_img/image_060906.png",
    "/workspace/test_img/image_076858.png"
  ],
  "report_to": "wandb",
  "logging_dir": "logs",
  "tracker_project_name": "CN_SMPL_sd15_v2",
  "push_to_hub": true,
  "hub_model_id": "yeq6x/CN_SMPL_sd15_v2",
  "hub_token": "",
  "allow_tf32": false,
  "enable_xformers_memory_efficient_attention": true,
  "enable_npu_flash_attention": false
}