diff --git a/examples/megatron/models/base_trainer.py b/examples/megatron/models/base_trainer.py
index 5151ee72..7dc10c49 100644
--- a/examples/megatron/models/base_trainer.py
+++ b/examples/megatron/models/base_trainer.py
@@ -48,7 +48,7 @@ def setup(self):
 
         self.args = get_args()
 
-        print(f"value trainer loading : {self.args.load}")
+        print(f"{self.name} trainer loading : {self.args.load}")
 
         self.model_type = ModelType.encoder_or_decoder
         self.tokenizer = get_tokenizer()