使用购买的万能模型报错，求大佬帮忙看看问题在哪。

koishi__

星级打分

• 1
• 2
• 3
• 4
• 5

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

使用群内买的《liae结构的迪丽热巴万能模型》再训练的时候报错，麻烦大佬指导一下是什么问题，要怎么解决，谢谢了！附报错图

xyqttacr

显存爆了

koishi__

xyqttacr 发表于 2024-5-2 17:48
显存爆了

麻烦指导一下要怎么解决。还是说用不了这个丹？   

天光脑震荡

把现在这块显卡卖了，换一块2080Ti22GB回来就行了

planenalp

batch_size 一点点降分别测试（8-7-6-5-4-3-2-1），直到能运行为止

koishi__

planenalp 发表于 2024-5-2 18:17
batch_size 一点点降分别测试（8-7-6-5-4-3-2-1），直到能运行为止

谢谢，我试试

nde2024

我的显卡是3060 12gb，曾经按原作者的配方(res:224, WF, archi:liae-udt, ae_dims:512, e_dims:64, d_dims:64, d_mask_dims:32, eyes_mouth_prio:N, blur_out_mask:Y, uniform_yaw:Y, lr_dropout:Y, batch:8. Others by default.)BS没开8，改开4

尝试练一个liae-udt架构的底单,不开 预训练 直接 OOM， 开 预训练 练一晚上OOM。

3060Ti的显存好像是8GB的，练起来估计更加困难。

后来我重建模型

resolution: 256                        ==
==             face_type: wf                         ==
==     models_opt_on_gpu: True                       ==
==                 archi: liae-ud                    ==
==               ae_dims: 320                        ==
==                e_dims: 96                         ==
==                d_dims: 96                         ==
==           d_mask_dims: 32      


就没有再OOM了。

dotahudie

batch_size 调成4试试