guided-diffusion
guided-diffusion copied to clipboard
Fix resumed model training for Multi-GPUs
Although the poster had closed the issue related to this https://github.com/openai/guided-diffusion/issues/23, I encountered the same issues when resuming training from a model checkpoint and solved it with OP's fix which is in this PR.