{
  "config": "./configs/train/reimple_second_stage_v2.yaml",
  "batch_size": 1,
  "motion_type": "dwpose",
  "is_second_stage": true,
  "freeze_controlnet": false,
  "use_HAP": false,
  "use_augchibi": false,
  "ref_aug": null,
  "motion_aug": null,
  "image_dataset_name": "TikTokImageDataset2",
  "video_dataset_name": "TikTokVideoDataset",
  "pretrained_vae_path": null,
  "pretrained_appearance_encoder_path": "./logs/from39_20240224_1st_freezectrl_temporalctrl/models/[AppearanceEncoder]_[Epoch=1250]_[Iter=17500]_[loss=0.0864].ckpt",
  "pretrained_controlnet_path": "./pretrained_models/ControlNet-v1-1/control_v11p_sd15_openpose_RenamedForMA.pth",
  "pretrained_model_path": null,
  "pretrained_unet_path": "./logs/from39_20240224_1st_freezectrl_temporalctrl/models/[UNet]_[Epoch=1250]_[Iter=17500]_[loss=0.0864].ckpt",
  "init_unet_lora": true,
  "init_appearance_encoder_lora": false,
  "load_unet_lora_weight": true,
  "load_ae_lora_weight": false,
  "use_learnable_taumap": false,
  "unet_lora_midup": false,
  "save_name": "SecondUnetloraTctrlv2_fromEpoch1250from39",
  "n_epochs": 90,
  "n_iters": 100000,
  "size": null,
  "data_root_dir": "./DATA/TikTok",
  "HAP_root_dir": "./DATA/HAP/train",
  "augchibi_root_dir": "./DATA/sd_augchibi",
  "save_root_dir": "./logs",
  "save_model_iter_freq": 2500,
  "validation_step": 2500,
  "validation_step_lst": [
    3
  ],
  "seed": 42,
  "num_workers": 2,
  "use_temporal_controlnet": true,
  "save_dir": "./logs/20240405_SecondUnetloraTctrlv2_fromEpoch1250from39",
  "sample_save_dir": "./logs/20240405_SecondUnetloraTctrlv2_fromEpoch1250from39/samples",
  "model_save_dir": "./logs/20240405_SecondUnetloraTctrlv2_fromEpoch1250from39/models",
  "tb_save_dir": "./logs/20240405_SecondUnetloraTctrlv2_fromEpoch1250from39/tb",
  "args_save_path": "./logs/20240405_SecondUnetloraTctrlv2_fromEpoch1250from39/args.json",
  "config_save_path": "./logs/20240405_SecondUnetloraTctrlv2_fromEpoch1250from39/config.yaml",
  "n_gpus": 2,
  "DDP": true,
  "num_processes": 2,
  "local_rank": 0
}