Clarify print.

cdf0a5d4 · Jared Casper · 96d19aae · cdf0a5d4
Commit cdf0a5d4 authored Feb 18, 2022 by Jared Casper
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

megatron/checkpointing.py megatron/checkpointing.py +1 -1

No files found.
--- a/megatron/checkpointing.py
+++ b/megatron/checkpointing.py
@@ -397,7 +397,7 @@ def load_args_from_checkpoint(args, load_arg='load'):
            checkpoint_value = getattr(checkpoint_args, arg_name, None)

        if checkpoint_value is not None:
-            print(f"Setting {arg_name} to {checkpoint_value}")
+            print(f"Setting {arg_name} to {checkpoint_value} from checkpoint")
            setattr(args, arg_name, checkpoint_value)

    _set_arg('num_layers')