Skip to content

GitLab

Commit cdf0a5d4 authored by Jared Casper's avatar Jared Casper
Browse files

Clarify print.

parent 96d19aae
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
megatron/checkpointing.py
View file @ cdf0a5d4
...@@ -397,7 +397,7 @@ def load_args_from_checkpoint(args, load_arg='load'): ...@@ -397,7 +397,7 @@ def load_args_from_checkpoint(args, load_arg='load'):
checkpoint_value = getattr(checkpoint_args, arg_name, None) checkpoint_value = getattr(checkpoint_args, arg_name, None)
if checkpoint_value is not 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) setattr(args, arg_name, checkpoint_value)
_set_arg('num_layers') _set_arg('num_layers')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment