Skip to content

GitLab

"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "6a05b274cc503276a4c1ac22a451df9184a9f761"
Commit 236a5ec0 authored by Lawrence McAfee's avatar Lawrence McAfee
Browse files

switched tensorboard memory logging from opt-out to opt-in

parent 24b7c3c0
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
megatron/arguments.py
View file @ 236a5ec0
...@@ -328,8 +328,8 @@ def _add_logging_args(parser): ...@@ -328,8 +328,8 @@ def _add_logging_args(parser):
action='store_true', action='store_true',
help='If set, write validation perplexity to ' help='If set, write validation perplexity to '
'tensorboard.') 'tensorboard.')
group.add_argument('--no-log-memory-to-tensorboard', group.add_argument('--log-memory-to-tensorboard',
action='store_false', action='store_true',
help='Disable memory logging to tensorboard.', help='Disable memory logging to tensorboard.',
dest='log_memory_to_tensorboard') dest='log_memory_to_tensorboard')
......
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