File size: 2,041 Bytes
fafe050 |
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 39 40 41 42 43 44 45 46 47 48 49 50 |
{
"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
} |