[refine] update train.py (#2262)

remove unnecessary variable names
wenet-e2e · Dec 26, 2023 · d427928 · d427928
1 parent 5fed2c4
commit d427928
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/wenet/bin/train.py b/wenet/bin/train.py
@@ -76,7 +76,7 @@ def main():
  tokenizer = init_tokenizer(configs)
 
  # Init env for ddp OR deepspeed
- world_size, local_rank, rank = init_distributed(args)
+ _, _, rank = init_distributed(args)
 
  # Get dataset & dataloader
  train_dataset, cv_dataset, train_data_loader, cv_data_loader = \