DFUD-wf-192 100W预训练+240W迭代 小显存的福音

lwtmylove

自带的FFHQ数据集 训练100W
用了一部分明星数据集训练240W
8G显存BS能开到16
4G勉强应该也可以跑的动


================== Model Summary ==================
==                                               ==
==            Model name: 192 HD-UD_SAEHD        ==
==                                               ==
==     Current iteration: 2405388                ==
==                                               ==
==---------------- Model Options ----------------==
==                                               ==
==            resolution: 192                    ==
==             face_type: wf                     ==
==     models_opt_on_gpu: True                   ==
==                 archi: df-ud                  ==
==               ae_dims: 256                    ==
==                e_dims: 64                     ==
==                d_dims: 64                     ==
==           d_mask_dims: 22                     ==
==       masked_training: True                   ==
==             eyes_prio: False                  ==
==           uniform_yaw: False                  ==
==            lr_dropout: n                      ==
==           random_warp: False                  ==
==             gan_power: 0.0                    ==
==       true_face_power: 0.0                    ==
==      face_style_power: 0.0                    ==
==        bg_style_power: 0.0                    ==
==               ct_mode: none                   ==
==              clipgrad: False                  ==
==              pretrain: False                  ==
==       autobackup_hour: 0                      ==
== write_preview_history: False                  ==
==           target_iter: 0                      ==
==           random_flip: False                  ==
==            batch_size: 16                     ==
==                                               ==
==----------------- Running On ------------------==
==                                               ==
==          DevICE index: 0                      ==
==                  Name: GeForce RTX 2070 SUPER ==
==                  VRAM: 8.00GB                 ==
==                                               ==
===================================================