Skip to content

Commit 0f7819a

Browse files
authored
Update main.py (pytorch#2117)
1 parent 10f0338 commit 0f7819a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/contrib/cifar10/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ def training(local_rank, config):
2424
manual_seed(config["seed"] + rank)
2525
device = idist.device()
2626

27-
logger = setup_logger(name="CIFAR10-Training", distributed_rank=local_rank)
27+
logger = setup_logger(name="CIFAR10-Training")
2828

2929
log_basic_info(logger, config)
3030

0 commit comments

Comments
 (0)