mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Global instead of local rank.

Browse Source
This commit is contained in:
SamuelGabriel 2021-06-09 19:11:58 +02:00 committed by GitHub
parent b7a568f065
commit 7c19c35d9f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -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: