Merge pull request #693 from SamuelGabriel/patch-1

Let only the _globally_ 0th rank write checkpoints in `train.py`

Author: rwightman

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: