LIAE_UD_WF224_398w RTM万能模型

yebingxue · 发表于 2022-2-26 13:52:31

星级打分

1
2
3
4
5

平均分:NAN 参与人数:0 我的评分:未评

感谢【夢生若汐顔】提供sc素材。
感谢【iperov】提供RTM训练方法及详细参数。

新手第一次训练模型，欢迎各位测试和指正。收大家10灵石，方便我更快的买镇坛之宝。

以下为模型参数：
================== Model Summary ===================
==                                              ==
==          Model name: new_SAEHD             ==
==                                              ==
==    Current iteration: 3979470                ==
==                                              ==
==---------------- Model Options -----------------==
==                                              ==
==          resolution: 224                   ==
==          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                   ==
==          uniform_yaw: False                ==
==          adabelief: True                   ==
==          lr_dropout: y                      ==
==          random_warp: False                ==
==          gan_power: 0.1                   ==
==    true_face_power: 0.0                   ==
==    face_style_power: 0.0                   ==
==       bg_style_power: 0.0                   ==
==             ct_mode: lct                   ==
==             clipgrad: False                ==
==             pretrain: False                ==
==    autobackup_hour: 0                      ==
== write_preview_history: False                ==
==          target_iter: 0                      ==
==          random_flip: False                ==
==          batch_size: 32                   ==
==          gan_warmup: 0.0                   ==
==    eyes_mouth_prio: True                   ==
==       gan_patch_size: 28                   ==
==             gan_dims: 32                   ==
==    random_src_flip: False                ==
==    random_dst_flip: False                ==
==       blur_out_mask: False                ==
==    random_hsv_power: 0.0                   ==
==                                              ==
==------------------ Running On ------------------==
==                                              ==
==       Device index: 0                      ==
==                Name: Tesla V100-PCIE-16GB-LS ==
==                VRAM: 13.66GB                ==
==       Device index: 1                      ==
==                Name: Tesla V100-PCIE-16GB-LS ==
==                VRAM: 13.66GB                ==
==       Device index: 2                      ==
==                Name: Tesla V100-PCIE-16GB-LS ==
==                VRAM: 13.66GB                ==
==       Device index: 3                      ==
==                Name: Tesla V100-PCIE-16GB-LS ==
==                VRAM: 13.66GB                ==
==                                              ==
====================================================

云盘内容如下，包含LIAE-UD模型和供deepfacelive使用的刘亦菲dfm模型。