jinyan 发表于 2024-1-27 14:54:20

AMP_320分辨率500W迭代次数分享



在400w迭代次数基础上,重置了迭代次数,又训练了100W次,建议用作男丹
训练步骤,先src-src 训练,删除 inter_dst 文件,然后src-dst训练,一个好丹最少训练百万迭代以上

================== Model Summary ===================
==                                                ==
==            Model name: willem-lillard-wf_AMP   ==
==                                                ==
==   Current iteration: 1010215               ==
==                                                ==
==---------------- Model Options -----------------==
==                                                ==
==    retraining_samples: False                   ==
==            resolution: 320                     ==
==             face_type: wf                      ==
==   models_opt_on_gpu: True                  ==
==               ae_dims: 448                     ==
==            inter_dims: 1024                  ==
==                e_dims: 102                     ==
==                d_dims: 102                     ==
==         d_mask_dims: 22                      ==
==          morph_factor: 0.5                     ==
==            preview_mf: 1                     ==
==       masked_training: True                  ==
==             eyes_prio: False                   ==
==            mouth_prio: False                   ==
==         uniform_yaw: False                   ==
==         loss_function: SSIM                  ==
==         blur_out_mask: True                  ==
==             adabelief: True                  ==
==            lr_dropout: n                     ==
==         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                     ==
==               ct_mode: none                  ==
==          random_color: False                   ==
==            clipgrad: True                  ==
==            use_fp16: False                   ==
==               cpu_cap: 8                     ==
==       preview_samples: 4                     ==
==    force_full_preview: False                   ==
==                  lr: 5e-05                   ==
==       autobackup_hour: 6                     ==
==          session_name: rw off                  ==
==   maximum_n_backups: 4                     ==
== write_preview_history: False                   ==
==         target_iter: 0                     ==
==       random_src_flip: False                   ==
==       random_dst_flip: True                  ==
==            batch_size: 8                     ==
==             gan_power: 0.0                     ==
==      gan_patch_size: 40                      ==
==            gan_dims: 16                      ==
==         gan_smoothing: 0.1                     ==
==             gan_noise: 0.0                     ==
==                                                ==
==------------------ Running On ------------------==
==                                                ==
==          Device index: 0                     ==
==                  Name: NVIDIA GeForce RTX 4090 ==
==                  VRAM: 21.57GB               ==
==                                                ==
====================================================


***本内容需购买可见***


页: [1]
查看完整版本: AMP_320分辨率500W迭代次数分享