From e604d2888390bd2c707bbbedb3f65d4efab4974f Mon Sep 17 00:00:00 2001 From: stcy Date: Fri, 30 Aug 2024 17:50:02 +0800 Subject: [PATCH] refine log output for trainer model --- examples/megatron/models/base_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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()