求助，尝试预训练遇到的问题

davidchars · 发表于 2022-1-17 11:10:28

星级打分

1
2
3
4
5

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

本帖最后由 davidchars 于 2022-1-18 09:55 编辑

把FFHQ数据集分成两份，分别放在src和dst里，然后正常训练，参数如下：

============================= Model Summary ==============================
==                                                                   ==
==          Model name: pretrain-liae-ud-256-256-64-64-20220113_SAEHD ==
==                                                                   ==
==    Current iteration: 500000                                     ==
==                                                                   ==
==--------------------------- Model Options ----------------------------==
==                                                                   ==
==          resolution: 256                                        ==
==          face_type: f                                           ==
==    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: True                                        ==
==          uniform_yaw: False                                        ==
==       blur_out_mask: False                                        ==
==          adabelief: True                                        ==
==          lr_dropout: n                                           ==
==          random_warp: True                                        ==
==    random_hsv_power: 0.0                                        ==
==    true_face_power: 0.0                                        ==
==    face_style_power: 0.0                                        ==
==       bg_style_power: 0.0                                        ==
==             ct_mode: lct                                        ==
==             clipgrad: False                                        ==
==             pretrain: False                                        ==
==    autobackup_hour: 8                                           ==
== write_preview_history: True                                        ==
==          target_iter: 500000                                     ==
==    random_src_flip: False                                        ==
==    random_dst_flip: False                                        ==
==          batch_size: 8                                           ==
==          gan_power: 0.0                                        ==
==       gan_patch_size: 32                                           ==
==             gan_dims: 16                                           ==
==                                                                   ==
==----------------------------- Running On -----------------------------==
==                                                                   ==
==       Device index: 0                                           ==
==                Name: NVIDIA GeForce GTX 1080 Ti                   ==
==                VRAM: 9.42GB                                     ==
==                                                                   ==
==========================================================================

预训练时的预览图如图1
然后正训的时候（张若昀替换基努李维斯）出现了问题，发现pred那一列跟dst是一样的，见图2

==========================================================================

现在问题已解决：

其实没有什么问题，多训练一些时间就正常了。看自动保存的结果，正训6W次的时候也正常。

总结：教程没问题，让子弹飞一会儿