白鹿80W迭代模型

928686799 · 发表于 2022-12-6 18:48:03

星级打分

1
2
3
4
5

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

本帖最后由 928686799 于 2023-9-7 09:13 编辑

底丹是论坛里下载的DF_224 百万预训练的模型，自己用白鹿的src跑了80W迭代，效果不错，底丹跑了半小时就有比较清晰的人脸了
================== Model Summary ===================
==                                              ==
==          Model name: Pretrained_SAEHD       ==
==                                              ==
==    Current iteration: 792940                ==
==                                              ==
==---------------- Model Options -----------------==
==                                              ==
==             use_fp16: False                ==
==          resolution: 224                   ==
==          face_type: wf                   ==
==    models_opt_on_gpu: True                   ==
==                archi: df-udt                ==
==             ae_dims: 256                   ==
==             e_dims: 64                   ==
==             d_dims: 64                   ==
==          d_mask_dims: 22                   ==
==    masked_training: True                   ==
==          mouth_prio: False                ==
==          uniform_yaw: False                ==
==       blur_out_mask: False                ==
==          adabelief: True                   ==
==          lr_dropout: n                      ==
==       loss_function: SSIM                   ==
==          random_warp: False                ==
==    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: lct                   ==
==       random_color: False                ==
==             clipgrad: False                ==
==             pretrain: False                ==
==             cpu_cap: 8                      ==
==    preview_samples: 4                      ==
== force_full_preview: False                ==
==                   lr: 5e-05                ==
==       session_name: SESSION NAME          ==
==    autobackup_hour: 0                      ==
==    maximum_n_backups: 24                   ==
== write_preview_history: False                ==
==          target_iter: 1000000                ==
== retraining_samples: False                ==
==    random_src_flip: False                ==
==    random_dst_flip: False                ==
==          batch_size: 8                      ==
==          gan_power: 0.0                   ==
==       gan_patch_size: 28                   ==
==             gan_dims: 16                   ==
==       gan_smoothing: 0.1                   ==
==          gan_noise: 0.0                   ==
==    eyes_mouth_prio: False                ==
==                                              ==
==------------------ Running On ------------------==
==                                              ==
==       Device index: 0                      ==
==                Name: NVIDIA GeForce RTX 2070 ==
==                VRAM: 6.57GB                ==
==                                              ==
=================================