Skip to content

GitLab

Commit 954f0725 authored by Raul Puri's avatar Raul Puri
Browse files

addressed printing

parent d28527d5
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
megatron/arguments.py
View file @ 954f0725
...@@ -53,9 +53,10 @@ def parse_args(extra_args_provider=None, defaults={}, ...@@ -53,9 +53,10 @@ def parse_args(extra_args_provider=None, defaults={},
# arguments that are passed to the program. We check this by # arguments that are passed to the program. We check this by
# ensuring the arg is set to None. # ensuring the arg is set to None.
if getattr(args, key) is not None: if getattr(args, key) is not None:
print('WARNING: overriding default arguments for {key}:{v} with \ if args.rank <= 0:
{key}:{v2}'.format(key=key, v=defaults[key], print('WARNING: overriding default arguments for {key}:{v} \
v2=getattr(args, key)) with {key}:{v2}'.format(key=key, v=defaults[key],
v2=getattr(args, key))
else: else:
setattr(args, key, defaults[key]) setattr(args, key, defaults[key])
......
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