egs/os_avatar/secc_img2plane_torso.yaml

base_config:
  - ./secc_img2plane.yaml

task_cls: tasks.os_avatar.secc_img2plane_torso_task.SECC_Img2PlaneEG3D_TorsoTask
torso_ref_segout_mode: torso # torso | torso_with_bg | person | full (person_with_bg)

lr_g: 0.00001

weight_fuse: true

start_adv_iters: 40000
max_updates: 10_0000 # 25_0000
lambda_th1kh_mv_adv: 0.001
add_ffhq_singe_disc: false
lambda_ffhq_mv_adv: 0.002 # enable when add_ffhq_singe_disc is True
lambda_mse: 1.0
init_from_ckpt: '' # checkpoints/0725_img2planes/secc_img2plane_torso | can be either a secc_img2plane or a secc_img2plane_torso ckpt
reload_head_ckpt: '' # checkpoints/0804_secc2plane/secc_img2plane_lap0.1_blink0.05_run2 | will override the secc_img2plane from init_from_ckpt and be reloaded during training


fuse_with_deform_source: false # fuse source会有严重的artifact
lam_occlusion_2_reg_l1: 0.0 # 0.001
torso_occlusion_reg_unmask_factor: 0.3
lam_occlusion_weights_entropy: 0.001 # 0.0001

lam_occlusion_reg_l1: 0.00 # 设置成0.02导致脸部和torso都有色差，并且摇头晃脑时只有脖子动，身体不太动，不真实。
occlusion_fuse: true
torso_kp_num: 4
htbsr_head_weight_fuse_mode: v2
htbsr_head_threshold: 0.9
torso_model_version: v2