训练自定义稳定扩散模型时如何避免内存耗尽错误?

问题描述 投票:0回答:1

enter image description here

我在训练自定义 LCM-LoRA 模型时遇到此错误。 我使用 347 个自定义图像来训练它。 我尝试使用 2 个 RTX-4090 GPU,但遇到了同样的错误。 我是否必须增加 GPU 数量或使用其他 GPU 类型? 我该如何处理这个错误? 谢谢

python deep-learning pytorch artificial-intelligence stable-diffusion
1个回答
0
投票

这可能是由于几个原因造成的,但我建议您首先尝试减小批量大小

© www.soinside.com 2019 - 2024. All rights reserved.