在使用批量大小为 1 的训练过程中内存不足
作者: KIMGEONUNG创建于 2023年1月31日更新于 2025年2月27日
感谢您的出色工作。 现在, 我正在尝试使用 RTX 3090 (24GB) 运行训练代码。 但即使批量大小为 1, 我也遇到了内存不足异常, 这实在是很奇怪… 与您在 README 中写的指示相比, 我运行代码的唯一区别是批量大小。 这是正常结果吗? 或者有什么方法可以减少所需的内存吗?
内容来源: timothybrooks/instruct-pix2pix
感谢您的出色工作。 现在, 我正在尝试使用 RTX 3090 (24GB) 运行训练代码。 但即使批量大小为 1, 我也遇到了内存不足异常, 这实在是很奇怪… 与您在 README 中写的指示相比, 我运行代码的唯一区别是批量大小。 这是正常结果吗? 或者有什么方法可以减少所需的内存吗?
内容来源: timothybrooks/instruct-pix2pix