From 7c19c35d9f1185a2cf017b33a98f1cd67ab30785 Mon Sep 17 00:00:00 2001
From: SamuelGabriel <Samuel.g.mueller@web.de>
Date: Wed, 9 Jun 2021 19:11:58 +0200
Subject: [PATCH] Global instead of local rank.

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 4264a164..f1c1581e 100755
--- a/train.py
+++ b/train.py
@@ -561,7 +561,7 @@ def main():
     best_epoch = None
     saver = None
     output_dir = None
-    if args.local_rank == 0:
+    if args.rank == 0:
         if args.experiment:
             exp_name = args.experiment
         else: