diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp
index cbb0b3b8a..85453e084 100644
Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp differ
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
index a3109aa97..aa73a31a9 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@@ -1135,10 +1135,10 @@ def train_one_epoch(
                     )
             
             if wb is not None and rank == 0:
-                wb.log({"train/loss": loss_info["loss"]*numel/params.world_size})
-                wb.log({"train/simple_loss": loss_info["simple_loss"]*numel/params.world_size})
-                wb.log({"train/pruned_loss": loss_info["pruned_loss"]*numel/params.world_size})
-                wb.log({"train/ctc_loss": loss_info["ctc_loss"]*numel/params.world_size})
+                wb.log({"train/loss": loss_info["loss"]*numel})
+                wb.log({"train/simple_loss": loss_info["simple_loss"]*numel})
+                wb.log({"train/pruned_loss": loss_info["pruned_loss"]*numel})
+                wb.log({"train/ctc_loss": loss_info["ctc_loss"]*numel})
 
 #if batch_idx % params.valid_interval == 0 and not params.print_diagnostics:
     logging.info("Computing validation loss")