大神能帮忙看下这个代码报错是什么原因呢？

qianzhi99 · 发表于 2023-11-15 21:53:03

13500025847 · 发表于 2023-11-16 02:49:37

到群里问更快有人回复啊

13500025847 · 发表于 2023-11-16 02:50:08

中毒了

五折 · 发表于 2023-11-16 13:15:49

第四行，OOM
显存不够

qianzhi99 · 发表于 2023-11-16 15:47:27

五折发表于 2023-11-16 13:15
第四行，OOM
显存不够

大神，我电脑显存8g 这个还不够吗

那这种迭代了那么多次的模型都用不了了嘛

若苗瞬 · 发表于 2023-11-16 16:03:12

qianzhi99 发表于 2023-11-16 15:47
大神，我电脑显存8g 这个还不够吗那这种迭代了那么多次的模型都用不了了嘛 ...

关闭 adabelief，
或者调小 batch size 呀。

PS：对于AI类，8G显存确实挺少的。

五折 · 发表于 2023-11-16 18:34:05

qianzhi99 发表于 2023-11-16 15:47
大神，我电脑显存8g 这个还不够吗那这种迭代了那么多次的模型都用不了了嘛 ...

你这个丹Liae-ud，Liae本身比DF重，参数没带t就更重，开了Lr和ct会提高占用

虽然我也觉得8G显存不应该BS1都开不了。。。。

优化器别放在GPU 关闭Lr和ct再试试

qianzhi99 · 发表于 2023-11-16 23:58:12

谢谢大神确实都关了就OK了

黑羽毛 · 发表于 2023-11-17 08:33:37

五折发表于 2023-11-16 18:34
你这个丹Liae-ud，Liae本身比DF重，参数没带t就更重，开了Lr和ct会提高占用

虽然我也觉得8G显存不应该BS ...

学到了，我有一个丹也是开始用不了，使用你的方法后就可以跑了，强

nero · 发表于 2023-11-18 01:29:41

减少resolution、ae_dims、e_dims和d_dims的值。
减小batch_size