Removed extra dist barrier

quic-meetkuma · quic-meetkuma · commit 998f7bb59cbb · 2025-04-10T17:11:57.000+05:30
Signed-off-by: Meet Patel &lt;quic_meetkuma@quicinc.com&gt;
diff --git a/QEfficient/finetune/utils/train_utils.py b/QEfficient/finetune/utils/train_utils.py
@@ -292,7 +292,6 @@ def train(
                 eval_ppl, eval_epoch_loss, temp_val_loss, temp_step_perplexity = evaluation(
                     model, train_config, eval_dataloader, local_rank, tokenizer, device
                 )
-                dist.barrier()
                 if local_rank == 0:
                     tensorboard_updates.add_scalars("loss", {"eval": eval_epoch_loss}, total_train_steps)
 

Original file line number	Diff line number	Diff line change
`@@ -292,7 +292,6 @@ def train(`
`292`	`292`	`eval_ppl, eval_epoch_loss, temp_val_loss, temp_step_perplexity = evaluation(`
`293`	`293`	`model, train_config, eval_dataloader, local_rank, tokenizer, device`
`294`	`294`	`)`
`295`		`- dist.barrier()`
`296`	`295`	`if local_rank == 0:`
`297`	`296`	`tensorboard_updates.add_scalars("loss", {"eval": eval_epoch_loss}, total_train_steps)`
`298`	`297`