[SAEHD] [128 WF] [预训练] [小显存] 4.8万亚洲人脸 100万迭代

4.8万 亚洲人脸数据（512分辨率）  迭代100w+的预训练模型。

我用的batch_size是4（占用显存4.64G），

如果再减小batch_size，4G卡 甚至2G卡 都没有问题。

参数如下：
=============== Model Summary ===============
==                                         ==
==            Model name: pretrained_SAEHD ==
==                                         ==
==     Current iteration: 1022731           ==
==                                         ==
==————- Model Options ————-==
==                                         ==
==            resolution: 128              ==
==             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             ==
==       eyes_mouth_prio: False            ==
==           uniform_yaw: True             ==
==         blur_out_mask: False            ==
==             adabelief: True             ==
==            lr_dropout: n                ==
==           random_warp: True             ==
==       true_face_power: 0.0              ==
==      face_style_power: 0.0              ==
==        bg_style_power: 0.0              ==
==               ct_mode: rct              ==
==              clipgrad: True             ==
==              pretrain: False            ==
==       autobackup_hour: 5                ==
== write_preview_history: False            ==
==           target_iter: 0                ==
==       random_src_flip: False            ==
==       random_dst_flip: True             ==
==            batch_size: 4                ==
==             gan_power: 0.0              ==
==        gan_patch_size: 16               ==
==              gan_dims: 16               ==
==                                         ==
==————– Running On —————==
==                                         ==
==          Device index: 0                ==
==                  Name: GeForce RTX 2060 ==
==                  VRAM: 4.64GB           ==
==                                         ==
=============================================

换脸模型下载链接，提取码需要点击下方付费查看：https://pan.baidu.com/s/1ch21jR7GnjFwunohy__SPA