Error: 2 root error(s) found.

cqlhj · 发表于 2022-7-28 23:59:35

星级打分

1
2
3
4
5

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

机器配置:
CPU:Intel E5-2673v3
内存:16G
硬盘:480G
显卡:RTX3090
系统:Win10 x64
软件版本

eepFaceLab_NVIDIA_RTX3000_servies_build_11_20_2021
缺省文件夹
操作如下:
2) extract images from video data_src.bat
3) extract images from video data_dst FULL FPS.bat
4) data_src faceset extract.bat
5) data_dst faceset extract.bat
6) train SAEHD.bat
参数及输出信息如下:
================== Model Summary ===================
==                                              ==
==          Model name: lhj_SAEHD             ==
==                                              ==
==    Current iteration: 0                      ==
==                                              ==
==---------------- Model Options -----------------==
==                                              ==
==          resolution: 224                   ==
==          face_type: wf                   ==
==    models_opt_on_gpu: True                   ==
==                archi: liae-ud                ==
==             ae_dims: 512                   ==
==             e_dims: 64                   ==
==             d_dims: 64                   ==
==          d_mask_dims: 32                   ==
==    masked_training: True                   ==
==    eyes_mouth_prio: False                ==
==          uniform_yaw: True                   ==
==       blur_out_mask: False                ==
==          adabelief: True                   ==
==          lr_dropout: y                      ==
==          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: none                   ==
==             clipgrad: False                ==
==             pretrain: False                ==
==    autobackup_hour: 0                      ==
== 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: 28                   ==
==             gan_dims: 16                   ==
==                                              ==
==------------------ Running On ------------------==
==                                              ==
==       Device index: 0                      ==
==                Name: NVIDIA GeForce RTX 3090 ==
==                VRAM: 21.16GB                ==
==                                              ==
====================================================
Starting. Press "Enter" to stop training and save model.
Trying to do the first iteration. If an error occurs, reduce the model parameters.
!!!
Windows 10 users IMPORTANT notice. You should set this setting in order to work correctly.
https://i.imgur.com/B7cmDCB.jpg
!!!
You are training the model from scratch. It is strongly recommended to use a pretrained model to speed up the training and improve the quality.

Error: 2 root error(s) found.
  (0) Resource exhausted: failed to allocate memory

..............
(1) Resource exhausted: failed to allocate memory

请指教,该怎么修改?
感谢!