[已解决] train SAEHD.bat Loading samples: 后无反应

sowole

星级打分

• 1
• 2
• 3
• 4
• 5

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

问题经QQ群里@帅哥 帮忙解决：设置虚拟内存即可解决。

本人新手，电脑配置：
4090 N卡，
i139K
32G内存

按照滚石老大的6) train SAEHD.bat 设置，到最后一步就卡在Loading samples:后就无反应。
ps: 6) train Quick96.bat 可以跑起来
但 6) train SAEHD.bat  不行

以下是我执行步骤：

Running trainer.

[new] No saved models found. Enter a name of a new model :
new

Model first run.

Choose one or several GPU idxs (separated by comma).

[CPU] : CPU
  [0] : NVIDIA GeForce RTX 4090

[0] Which GPU indexes to choose? :
0

[0] Autobackup every N hour ( 0..24 ?:help ) :
0
[n] Write preview history ( y/n ?:help ) :
n
[0] Target iteration :
0
[n] Flip SRC faces randomly ( y/n ?:help ) :
n
[y] Flip DST faces randomly ( y/n ?:help ) :
y
[8] Batch_size ( ?:help ) :
8
[128] Resolution ( 64-640 ?:help ) :
128
[f] Face type ( h/mf/f/wf/head ?:help ) : wf
wf
[liae-ud] AE architecture ( ?:help ) :
liae-ud
[256] AutoEncoder dimensions ( 32-1024 ?:help ) :
256
[64] Encoder dimensions ( 16-256 ?:help ) :
64
[64] Decoder dimensions ( 16-256 ?:help ) :
64
[22] Decoder mask dimensions ( 16-256 ?:help ) :
22
[y] Masked training ( y/n ?:help ) :
y
[n] Eyes and mouth priority ( y/n ?:help ) :
n
[n] Uniform yaw distribution of samples ( y/n ?:help ) :
n
[n] Blur out mask ( y/n ?:help ) :
n
[y] Place models and optimizer on GPU ( y/n ?:help ) :
y
[y] Use AdaBelief optimizer? ( y/n ?:help ) :
y
[n] Use learning rate dropout ( n/y/cpu ?:help ) :
n
[y] Enable random warp of samples ( y/n ?:help ) :
y
[0.0] Random hue/saturation/light intensity ( 0.0 .. 0.3 ?:help ) :
0.0
[0.0] GAN power ( 0.0 .. 5.0 ?:help ) :
0.0
[0.0] Face style power ( 0.0..100.0 ?:help ) :
0.0
[0.0] Background style power ( 0.0..100.0 ?:help ) :
0.0
[none] Color transfer for src faceset ( none/rct/lct/mkl/idt/sot ?:help ) :
none
[n] Enable gradient clipping ( y/n ?:help ) :
n
[n] Enable pretraining mode ( y/n ?:help ) :
n
Initializing models: 100%|###############################################################| 5/5 [00:01<00:00,  4.79it/s]
Loading samples: 100%|##############################################################| 654/654 [00:01<00:00, 415.44it/s]

ccctttccct

虚拟内存搞了没

ccctttccct

对了还有硬件加速开了没

胡起豪

直接设置个100G虚拟内存，具体操作详见百度

sowole

感谢群里的 @帅哥 大大
1，设置虚拟内存，设置办法:
     win11：设置-》系统信息-》高级系统设置-》高级-》性能-》设置-》高级-》程序-》更改-》去掉自动管理所有驱动器分页文件大小-》选择一个自己想要设置的盘符-》自定义大小-》设置-》重新电脑