关于训练过程的 GPU 内存使用情况
作者: zy-charon创建于 2022年12月11日更新于 2025年4月9日
你好,我之前已经重现了 DDPM 基本模型的实验。当我使用 128*128 尺寸的图片,并将 batchsize 设置为 2 时,GPU 内存使用量大约为 18G;当我使用你的代码时,同样的设置,GPU 的使用量大约为 5G,这是正常的吗,或者可能是我的设置出了问题?
内容来源: openai/guided-diffusion
你好,我之前已经重现了 DDPM 基本模型的实验。当我使用 128*128 尺寸的图片,并将 batchsize 设置为 2 时,GPU 内存使用量大约为 18G;当我使用你的代码时,同样的设置,GPU 的使用量大约为 5G,这是正常的吗,或者可能是我的设置出了问题?
内容来源: openai/guided-diffusion