训练SAEHD可以训练无法保存

津铁院张主任

星级打分

• 1
• 2
• 3
• 4
• 5

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

训练SAEHD可以训练你，但是保存报错，这里的memory error是磁盘空间不足还是显存不足

参数和报错图片如下


==================== Model Summary ====================
==                                                   ==
==            Model name: 192fdfud_SAEHD             ==
==                                                   ==
==     Current iteration: 5007848                    ==
==                                                   ==
==------------------ Model Options ------------------==
==                                                   ==
==            resolution: 192                        ==
==             face_type: f                          ==
==     models_opt_on_gpu: True                       ==
==                 archi: df-ud                      ==
==               ae_dims: 256                        ==
==                e_dims: 64                         ==
==                d_dims: 64                         ==
==           d_mask_dims: 22                         ==
==       masked_training: True                       ==
==           uniform_yaw: False                      ==
==            lr_dropout: y                          ==
==           random_warp: False                      ==
==             gan_power: 0.1                        ==
==       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_flip: False                      ==
==            batch_size: 4                          ==
==       random_src_flip: False                      ==
==       random_dst_flip: False                      ==
==        gan_patch_size: 24                         ==
==              gan_dims: 16                         ==
==                                                   ==
==------------------- Running On --------------------==
==                                                   ==
==          Device index: 0                          ==
==                  Name: AMD Radeon(TM) Graphics    ==
==                  VRAM: 7.54GB                     ==
==                                                   ==
=======================================================

haiou_f

原来这个显卡也能运行，你调低一下bs，把虚拟内存设置大一些试试

津铁院张主任

haiou_f 发表于 2023-2-12 21:33
原来这个显卡也能运行，你调低一下bs，把虚拟内存设置大一些试试

谢谢你，请问虚拟内存咋设置，没搜到

haiou_f

津铁院张主任 发表于 2023-2-12 21:36
谢谢你，请问虚拟内存咋设置，没搜到

如果是win10的话，鼠标右键单击“此电脑”-高级系统设置-高级-设置，进去后点“高级”，修改虚拟内存，建议找个空间比较大的盘
另外“高级”下的环境变量最好设置一下，temp文件夹和tmp文件夹的位置都放在虚拟内存所在盘

津铁院张主任

haiou_f 发表于 2023-2-12 22:13
如果是win10的话，鼠标右键单击“此电脑”-高级系统设置-高级-设置，进去后点“高级”，修改虚拟内存，建 ...

已经解决了，谢谢你！