Remove dangling barrier

2022-11-30 22:52:48 -07:00 · 2022-11-30 22:52:48 -07:00 · 55a555850d
parent 7102d313ac
commit 55a555850d
1 changed files with 0 additions and 1 deletions
--- a/trainer/diffusers_trainer.py
+++ b/trainer/diffusers_trainer.py
@ -957,7 +957,6 @@ def main():

    save_checkpoint(global_step)

-    torch.distributed.barrier()
    cleanup()

    print(get_gpu_ram())