这个loss怎么降不下来？

rrtcce · 发表于 2023-12-29 16:53:08

本帖最后由 rrtcce 于 2023-12-29 16:55 编辑

我用的 TMBDF 新工作流程来跑的

==                                                 ==
==------------------ Model Options ------------------==
==                                                 ==
==             use_fp16: False                   ==
==          resolution: 320                      ==
==          face_type: wf                      ==
==    models_opt_on_gpu: True                      ==
==                archi: liae-ud                   ==
==             ae_dims: 256                      ==
==             e_dims: 64                      ==
==             d_dims: 64                      ==
==          d_mask_dims: 22                      ==
==    masked_training: True                      ==
== retraining_samples: False                   ==
==          eyes_prio: False                   ==
==          mouth_prio: False                   ==
==          uniform_yaw: True                      ==
==       blur_out_mask: True                      ==
==          adabelief: True                      ==
==          lr_dropout: n                         ==
==       loss_function: SSIM                      ==
==          random_warp: True                      ==
==    random_hsv_power: 0.0                      ==
==    random_downsample: False                   ==
==       random_noise: False                   ==
==          random_blur: False                   ==
==          random_jpeg: False                   ==
==       random_shadow: none                      ==
==    background_power: 0.0                      ==
==    true_face_power: 0.0                      ==
==    face_style_power: 0.0                      ==
==       bg_style_power: 0.0                      ==
==             ct_mode: rct                      ==
==       random_color: False                   ==
==             clipgrad: True                      ==
==             pretrain: False                   ==
==             cpu_cap: 8                         ==
==    preview_samples: 1                         ==
== force_full_preview: False                   ==
==                   lr: 0.0                      ==
==       session_name: m5                      ==
==    autobackup_hour: 1                         ==
==    maximum_n_backups: 1                         ==
== write_preview_history: False                   ==
==          target_iter: 500000                   ==
==    random_src_flip: False                   ==
==    random_dst_flip: True                      ==
==          batch_size: 8                         ==
==          gan_power: 0.0                      ==
==       gan_patch_size: 32                      ==
==             gan_dims: 16                      ==
==       gan_smoothing: 0.1                      ==
==          gan_noise: 0.0                      ==
==                                                 ==

dst用的官方的dst rtm图集9G 的faceset.pak

微信图片_20231229165119.png