练丹速度好慢哟，这参数有什么好的建议吗

aixx

星级打分

• 1
• 2
• 3
• 4
• 5

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

batch_size从4到8到16，发现练丹速度并没提高
以下参数哪些特别影响练丹的速度
================== Model Summary ===================
==                                                ==
==            Model name: heejin_SAEHD            ==
==                                                ==
==     Current iteration: 2645644                 ==
==                                                ==
==---------------- Model Options -----------------==
==                                                ==
==            resolution: 320                     ==
==             face_type: wf                      ==
==     models_opt_on_gpu: True                    ==
==                 archi: liae-udt                ==
==               ae_dims: 384                     ==
==                e_dims: 96                      ==
==                d_dims: 96                      ==
==           d_mask_dims: 32                      ==
==       masked_training: True                    ==
==       eyes_mouth_prio: True                    ==
==           uniform_yaw: True                    ==
==         blur_out_mask: False                   ==
==             adabelief: True                    ==
==            lr_dropout: n                       ==
==           random_warp: True                    ==
==      random_hsv_power: 0.1                     ==
==       true_face_power: 0.0                     ==
==      face_style_power: 0.0                     ==
==        bg_style_power: 0.0                     ==
==               ct_mode: rct                     ==
==              clipgrad: True                    ==
==              pretrain: False                   ==
==       autobackup_hour: 2                       ==
== write_preview_history: False                   ==
==           target_iter: 0                       ==
==       random_src_flip: False                   ==
==       random_dst_flip: True                    ==
==            batch_size: 16                      ==
==             gan_power: 0.0                     ==
==        gan_patch_size: 40                      ==
==              gan_dims: 16                      ==
==              use_fp16: False                   ==
==            mouth_prio: True                    ==
==         loss_function: SSIM                    ==
==     random_downsample: False                   ==
==          random_noise: False                   ==
==           random_blur: False                   ==
==           random_jpeg: False                   ==
==         random_shadow: none                    ==
==      background_power: 0.0                     ==
==          random_color: False                   ==
==               cpu_cap: 8                       ==
==       preview_samples: 16                      ==
==    force_full_preview: False                   ==
==                    lr: 5e-05                   ==
==          session_name:                         ==
==     maximum_n_backups: 24                      ==
==    retraining_samples: True                    ==
==         gan_smoothing: 0.1                     ==
==             gan_noise: 0.0                     ==
==                                                ==
==------------------ Running On ------------------==
==                                                ==
==          Device index: 0                       ==
==                  Name: NVIDIA GeForce RTX 3090 ==
==                  VRAM: 21.17GB                 ==
==                                                ==
====================================================
Starting. Press "Enter" to stop training and save model.
[04:17:48][#2646883][2575ms][0.8872][0.5608]
[04:42:40][#2648136][2608ms][0.5751][0.4341]
[05:07:41][#2649388][3611ms][0.5092][0.3976]
[05:32:41][#2650649][3804ms][0.4724][0.3766]
[05:58:14][#2651910][3936ms][0.4421][0.3612]
[06:22:37][#2653144][3631ms][0.4276][0.3520]
[06:47:39][#2654410][3635ms][0.4071][0.3422]
[07:12:38][#2655674][3667ms][0.3967][0.3337]
[07:37:38][#2656934][3610ms][0.3817][0.3242]
[08:03:07][#2658196][4039ms][0.3762][0.3234]
[08:27:37][#2659435][3829ms][0.3665][0.3165]
[08:52:37][#2660698][4827ms][0.3579][0.3083]
[09:17:37][#2661962][3637ms][0.3532][0.3057]
[09:42:35][#2663226][3638ms][0.3441][0.3009]
[10:08:01][#2664492][4007ms][0.3403][0.2977]
[10:32:36][#2665722][3795ms][0.3335][0.2952]
[10:57:37][#2666975][3823ms][0.3293][0.2927]
[11:22:35][#2668240][3610ms][0.3243][0.2877]
[11:47:36][#2669507][3623ms][0.3232][0.2876]
[12:13:00][#2670774][4062ms][0.3173][0.2831]
[12:37:36][#2672019][3741ms][0.3129][0.2815]
[13:02:36][#2673280][3817ms][0.3109][0.2788]
[13:27:35][#2674547][3650ms][0.3079][0.2760]
[13:52:57][#2675812][4010ms][0.3046][0.2736]
[14:17:34][#2677058][2763ms][0.3023][0.2714]
[14:42:36][#2678324][3815ms][0.2988][0.2703]
[15:07:36][#2679589][3821ms][0.2959][0.2708]
[15:32:36][#2680850][3756ms][0.2947][0.2685]
[15:57:56][#2682115][4018ms][0.2904][0.2636]
[16:22:36][#2683363][3618ms][0.2883][0.2629]
[16:47:37][#2684629][4788ms][0.2883][0.2636]
[17:12:35][#2685894][3659ms][0.2854][0.2597]
[17:37:36][#2687159][3864ms][0.2833][0.2604]
[18:02:57][#2688396][2950ms][0.2813][0.2570]
[

若苗瞬

random_warp: True

eyes_mouth_prio: True

lr_dropout: n

额。。。

别一直随机扭曲，练够了改false吧。

眼睛嘴改成False会快些，
但是需要人工注意眼神有没有不对。

随机扭曲关后，可以开学习率丢弃。
随机扭曲关后，BS保持大没错。

Mrdeepfake

这速度其实已经够快了。硬件上影响训练速度的主要因素是显卡性能，英伟达显卡比其他牌子显卡性能更佳。软件上影响训练速度的主要因素是计算机算法（暂时不讨论这个），模型分辨率、训练素材数目、取脸精度等。看你模型 loss 值还在下降，还需要开随机扭曲继续训练。但眼、嘴优先选项建议你先关闭吧，跑太久对眼、嘴以外的部分学习不利。作为一种可替代方案，你其实可以把开启/关闭眼嘴优先训练交替进行的，对模型整体收敛更有利！还可以夹杂uniform yaw、LRD 的开/关等。

come3002

方案1：升级SRC
让大模型model，学更清晰的图片会让dfl学的更精准。提高效率。
（假设你的SRC是一团模糊，看不清下牙齿。炼再久也作品也难出出下牙齿）
具体的说。比如之前4k电视剧大量的远景模糊、根本看不清的。改为4k近景的清晰的SRC。
或增加 近景 清晰的src比例



左1 远景 看不清眉毛，左2 则能看到眉毛

以能看到眉毛、下牙齿齿缝，嘴唇纹理等为佳。

方案2：DST提升。若国内女性杂图难度大。
可以从樱花下手。
如1080p 樱花改为4k 高清的樱花。
就不上图了。

iyuandou

由衷佩服能忍受这么高的迭代时间

若苗瞬

iyuandou 发表于 2023-9-13 05:51
由衷佩服能忍受这么高的迭代时间

他320分辨率，BS16，这速度已经很快了。

htyy

这么低了还不开学习吗

s659982344

要不要再考虑加一张卡