【已搜无结果】训练SAEHD模型报错，求助大佬

bug008

星级打分

• 1
• 2
• 3
• 4
• 5

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

错误代码如下：


F tensorflow/c/logging.cc:43] HRESULT failed with 0x887a0001: dml_device_->GetDeviceRemovedReason()
F tensorflow/c/logging.cc:43] HRESULT failed with 0x887a0005: dml_device->CreateOperatorInitializer( ABSL_ARRAYSIZE(ops), ops, IID_PPV_ARGS(&initializer))


用的是五彩288做底丹进行训练，是显存（16G）爆了么，求助各位大佬




[n] 是否要重置迭代计器和LOSS图？ ( y/n  按英文?获取帮助信息 ) : n
[0] 几个小时备份一次 ( 0..24  按英文?获取帮助信息 ) : 1
[n] 保存预览历史记录 ( y/n  按英文?获取帮助信息 ) : n
[0] 目标迭代次数 : 0
[n] 随机翻转SRC人脸 ( y/n  按英文?获取帮助信息 ) : n
[y] 随机翻转DST人脸 ( y/n  按英文?获取帮助信息 ) : n
[16] 批量大小 (  按英文?获取帮助信息 ) : 8
[y] 训练遮罩内的人脸 Masked training ( y/n  按英文?获取帮助信息 ) : y
[n] 眼睛和嘴巴优先 Eyes and mouth priority ( y/n  按英文?获取帮助信息 ) : n
[n] 侧脸优化 Uniform yaw distribution of samples ( y/n  按英文?获取帮助信息 ) : n
[n] 模糊遮罩边缘 Blur out mask ( y/n  按英文?获取帮助信息 ) : n
[y] 将模型和优化器放在GPU上运行 Place models and optimizer on GPU ( y/n  按英文?获取帮助信息 ) : y
[y] 使用AdaBelief优化器? Use AdaBelief optimizer? ( y/n  按英文?获取帮助信息 ) : y
[n] 使用学习率下降 Use learning rate dropout ( n/y/cpu  按英文?获取帮助信息 ) : n
[y] 随机扭曲 Enable random warp of samples ( y/n  按英文?获取帮助信息 ) : y
[0.0] 随机色调/饱和度/光强度 ( 0.0 .. 0.3  按英文?获取帮助信息 ) : 0.0
[0.0] GAN强度 GAN power ( 0.0 .. 5.0  按英文?获取帮助信息 ) : 0.1
[36] gan补丁大小 GAN patch size ( 3-640  按英文?获取帮助信息 ) : 36
[16] GAN维度 GAN dimensions ( 4-512  按英文?获取帮助信息 ) : 16
[0.0] '真脸强度 True face' power. ( 0.0000 .. 1.0  按英文?获取帮助信息 ) : 0.001
[0.0] 人脸风格强度 Face style power ( 0.0..100.0  按英文?获取帮助信息 ) : 0.001
[0.0] 背景风格强度 Background style power ( 0.0..100.0  按英文?获取帮助信息 ) : 0.0
[none] 颜色转换模式 Color transfer for src faceset ( none/rct/lct/mkl/idt/sot  按英文?获取帮助信息 ) : none
[n] 启用梯度剪裁 Enable gradient clipping ( y/n  按英文?获取帮助信息 ) : y
[n] 启用预训练 ( y/n  按英文?获取帮助信息 ) : n

pasanonic

bs 调小
随即扭曲 GAN同时开冲突 ,gan如果你没备份建议开梯度


你都参数好乱 ,把教程看完

皮萝鼠的卷心菜

菜鸡认为大哥你教程都没看完蛤蛤， 把BS降低，GAN强度0，真脸强度0，人脸风格0就好了

bug008

菜鸡谢谢各位大佬

bobowxc

你这个参数开的有点杂啊，要是入手可以试试192或者224，成像很快，但是效果不好，适合新手期探索

cfmmx

将模型和优化器放在GPU上运行 Place models and optimizer on GPU ( y/n  按英文?获取帮助信息 ) :
n

crimoon

cfmmx 发表于 2024-8-16 09:55
将模型和优化器放在GPU上运行 Place models and optimizer on GPU ( y/n  按英文?获取帮助信息 ) :
n ...

为什么这个是选N呢

crimoon

我也是这个问题，请问解决了吗，同一参数能跑，然后跑了几个小时了突然报错2024-08-16 17:57:19.203541: F tensorflow/c/logging.cc:43] HRESULT failed with 0x887a0001: dml_device_->GetDeviceRemovedReason()
请按任意键继续. . .

crimoon

cfmmx 发表于 2024-8-16 09:55
将模型和优化器放在GPU上运行 Place models and optimizer on GPU ( y/n  按英文?获取帮助信息 ) :
n ...

我电脑放着自己跑跑了几个小时突然出现了一下报错，就是说明我是顶着临界在跑，所以这个开成N提高裕度，请问我理解的对吗？

htyy

新手就别乱开参数，自己看看其他训练步骤的帖子，降低bs