Mirrors
/
waifu-diffusion
mirror of https://github.com/harubaru/waifu-diffusion.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Remove dangling barrier

This commit is contained in:
Anthony Mercurio 2022-11-30 22:52:48 -07:00 committed by GitHub
parent 7102d313ac
commit 55a555850d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
trainer/diffusers_trainer.py
View File

@ -957,7 +957,6 @@ def main():
save_checkpoint(global_step) save_checkpoint(global_step)
torch.distributed.barrier()
cleanup() cleanup()
print(get_gpu_ram()) print(get_gpu_ram())