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