在只有一个GPU资源下可以训练吗？

[wwj723452017]

作者你好，readme里面写的是“We use 4A40 GPUs and take 48 hours training the diffusion model, and use 2RTX3090 GPUs and take 3 hours training the guided classifier.”
想请问在只有一个GPU下可以训练模型吗，是否通过修改超参数实现?

[cnulab]

你好！
只有一个GPU扩散模型可能训练不了，后面的步骤比如采样以及训练RealNet都是可以的。你如果只在比如MVTec或者VisA这样的数据集上实验，可以直接下载我提供的模型（diffusion和guided classifier）。