小白求解

ACG · 发表于 2025-2-10 16:57:20

wtxx8888 · 发表于 2025-2-10 17:52:58

OOM炸显存了，模型目前的参数配置，你显卡带不动。

ACG · 发表于 2025-2-10 18:39:29

本帖最后由 ACG 于 2025-2-10 18:41 编辑

wtxx8888 发表于 2025-2-10 17:52
OOM炸显存了，模型目前的参数配置，你显卡带不动。

我都是按默认来啊，而且我的显存大于50，没想到连4070都带不动了

奸商 · 发表于 2025-2-10 20:16:15

ACG 发表于 2025-2-10 18:39
我都是按默认来啊，而且我的显存大于50，没想到连4070都带不动了

啥叫显存大于50
你的4070只有5.28G可用显存
当模型尺寸超过这个值，显存就会溢出。
决定模型尺寸的是模型参数大小和BS大小，所以一是要小泽小参数模型，而是要设置合理的BS大小

ACG · 发表于 2025-2-10 21:54:24

奸商发表于 2025-2-10 20:16
啥叫显存大于50
你的4070只有5.28G可用显存
当模型尺寸超过这个值，显存就会溢出。

解决了，感谢

ACG · 发表于 2025-2-10 23:12:53

本帖最后由 ACG 于 2025-2-11 00:36 编辑

奸商发表于 2025-2-10 20:16
啥叫显存大于50
你的4070只有5.28G可用显存
当模型尺寸超过这个值，显存就会溢出。

有些模型的显存太大，我把分辨率调跟BS调为256和为4，还是超出了我的显存，还有什么办法吗

h756154515 · 发表于 2025-2-11 23:29:09

是不是gan power太高了也会爆显存

ubiqq · 发表于 2025-2-12 23:41:37

希望有帮助：
防止爆显存
1、选择三围合适的底模
2、一定记得开启Enablegradient clipping
3、适当减小batch_size
4、可以关闭Place models and optimizer on GPU 和 Use AdaBelief optimizer（这个我也没试着关过，应该是训练会变慢）
5、关闭 gan、true face、background style power

ACG · 发表于 2025-2-13 09:26:47

h756154515 发表于 2025-2-11 23:29
是不是gan power太高了也会爆显存

感谢

ACG · 发表于 2025-2-13 09:27:19

ubiqq 发表于 2025-2-12 23:41
希望有帮助：
防止爆显存
1、选择三围合适的底模

感谢

小白求解

评分

浏览过的版块

最佳新人

热心会员

咸鱼勋章