请教大佬，训练卡住了

smdongxi1

按照官网上大佬的做了环境部署，安装了CUDA和适配驱动了，虚拟内存设置在C盘24000-128000，但是运行SAEHD后，每次都是加载到100%就不动了，网上搜了可能是虚拟内存问题，但已经设置这么大了为什么还是卡住了。data_dst文件夹里没有目标图片，data_dst\aligned里有截出来的人脸。data_src文件夹里有源图片，data_src\aligned里有截取出来的图片。但感觉不是这个差异导致的。



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

[CPU] : CPU
  [0] : NVIDIA GeForce RTX 5060 Laptop GPU

[0] Which GPU indexes to choose? : 0
0

[2] Autobackup every N hour ( 0..24 ?:help ) :
2
[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
[4] Batch_size ( ?:help ) :
4
[256] Resolution ( 64-640 ?:help ) :
256
[wf] Face type ( h/mf/f/wf/head ?:help ) :
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
[rct] Color transfer for src faceset ( none/rct/lct/mkl/idt/sot ?:help ) :
rct
[y] Enable gradient clipping ( y/n ?:help ) :
y
[n] Enable pretraining mode ( y/n ?:help ) :
n
Initializing models: 100%|###############################################################| 5/5 [00:03<00:00,  1.44it/s]
Loading samples: 100%|##############################################################| 197/197 [00:00<00:00, 320.98it/s]
Loading samples: 100%|############################################################| 2795/2795 [00:08<00:00, 347.76it/s]

五折

原版DFL对50系显卡的支持有问题
试试Ice的1.818版本

smdongxi1

五折 发表于 2025-8-22 00:44
原版DFL对50系显卡的支持有问题
试试Ice的1.818版本

收到，谢谢大佬