koishi__ 发表于 2024-5-2 17:39:33

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

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

















xyqttacr 发表于 2024-5-2 17:48:20

显存爆了

koishi__ 发表于 2024-5-2 18:07:03

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

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

天光脑震荡 发表于 2024-5-2 18:16:45

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

planenalp 发表于 2024-5-2 18:17:47

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

koishi__ 发表于 2024-5-2 18:24:33

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

谢谢，我试试

nde2024 发表于 2024-5-3 16:52:28

本帖最后由 nde2024 于 2024-5-3 16:53 编辑

我的显卡是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 发表于 2024-5-8 06:01:43

batch_size 调成4试试
页: [1]
查看完整版本: 使用购买的万能模型报错，求大佬帮忙看看问题在哪。