训练时OOM怎么解决

[uptodiff]

笔记本电脑显卡是1050ti，训练模型时出现OOM的报错信息，这个应该怎么办呢

[shaohelv]

OOM通常是因为显存不足。

1. 你的 图片大小设置是多少呢？
2. 1050ti确实有点不够，建议换用1080ti或titanx。
3. 笔记本跑，也还是吃力。
4. 建议到网上租高端显卡。

我就是租显卡跑的。一直在用“易学智能”这个平台，打比赛、换脸，断断续续用了快半年了。黑科技满满。可以去用一下，最低可只购买一小时就能体验。