训练出现OOM 求助各位大佬

Tdeepface · 发表于 2021-6-1 15:34:49

星级打分

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

本帖最后由 Tdeepface 于 2021-6-1 15:36 编辑

问题如图参数均为默认

1048809231 · 发表于 2021-6-1 21:37:25

bs调小， Place models and optimizer on GPU这项可以关了，Use AdaBelief optimizer这项也可以关。总之一项一项的关，直到能跑

haiou_f · 发表于 2021-6-1 16:29:31

调低bs值就行

阳小咩咩阳 · 发表于 2021-6-1 17:09:23

疯狂水丹

whl716694 · 发表于 2021-6-1 17:43:03

bs值从2开始试

moweijun · 发表于 2021-6-2 12:30:41

Batch_size调小，Place models and optimizer on GPU和Use AdaBelief optimizer都关了，应该就能跑了，要是这都不行，换丹吧

1996115 · 发表于 2021-6-2 21:09:04

全默认肯定不行啊

shen521 · 发表于 2021-6-3 09:47:30

他们说的都不对，要换显卡才是根本解决办法， rtx 3080ti 少喝点奶茶就可以了

阳小咩咩阳 · 发表于 2021-6-3 10:04:52

路过看看

demonash · 发表于 2021-6-7 12:41:51

谢谢分享~~~！

		自动登录	找回密码
密码			立即注册（仅限QQ邮箱）