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

236a5ec0 · Lawrence McAfee · 24b7c3c0 · 236a5ec0
Commit 236a5ec0 authored Jul 12, 2021 by Lawrence McAfee
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

megatron/arguments.py megatron/arguments.py +2 -2

No files found.
--- a/megatron/arguments.py
+++ b/megatron/arguments.py
@@ -328,8 +328,8 @@ def _add_logging_args(parser):
                       action='store_true',
                       help='If set, write validation perplexity to '
                       'tensorboard.')
-    group.add_argument('--no-log-memory-to-tensorboard',
-                       action='store_false',
+    group.add_argument('--log-memory-to-tensorboard',
+                       action='store_true',
                       help='Disable memory logging to tensorboard.',
                       dest='log_memory_to_tensorboard')