用 SAEHD 訓練 CPU 問題

loueas · 发表于 2022-1-11 00:15:08

星级打分

1
2
3
4
5

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

本帖最后由 loueas 于 2022-1-12 00:31 编辑

如題,當用 SAEHD 訓練時 CPU 都會100%
GPU 都會在10~20%
我的顯卡 RTX3070
需要在哪做修改

補上數據
下面是我的設定資訊
我需要修改哪裡
================== Model Summary ===================
==                                              ==
==          Model name: 30_SAEHD             ==
==                                              ==
==    Current iteration: 0                      ==
==                                              ==
==---------------- Model Options -----------------==
==                                              ==
==          resolution: 64                   ==
==          face_type: mf                   ==
==    models_opt_on_gpu: True                   ==
==                archi: liae-ud                ==
==             ae_dims: 32                   ==
==             e_dims: 64                   ==
==             d_dims: 64                   ==
==          d_mask_dims: 22                   ==
==    masked_training: True                   ==
==    eyes_mouth_prio: True                   ==
==          uniform_yaw: True                   ==
==       blur_out_mask: False                ==
==          adabelief: True                   ==
==          lr_dropout: n                      ==
==          random_warp: False                ==
==    random_hsv_power: 0.0                   ==
==    true_face_power: 0.0                   ==
==    face_style_power: 0.0                   ==
==       bg_style_power: 0.0                   ==
==             ct_mode: rct                   ==
==             clipgrad: False                ==
==             pretrain: False                ==
==    autobackup_hour: 0                      ==
== write_preview_history: False                ==
==          target_iter: 0                      ==
==    random_src_flip: False                ==
==    random_dst_flip: True                   ==
==          batch_size: 2                      ==
==          gan_power: 0.0                   ==
==       gan_patch_size: 16                   ==
==             gan_dims: 16                   ==
==                                              ==
==------------------ Running On ------------------==
==                                              ==
==       Device index: 0                      ==
==                Name: NVIDIA GeForce RTX 3070 ==
==                VRAM: 5.34GB                ==
==                                              ==
====================================================

下圖是在跑時的效能狀況