Commit 236a5ec0 authored by Lawrence McAfee's avatar Lawrence McAfee
Browse files

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

parent 24b7c3c0

megatron/arguments.py

+2 −2
Original line number Diff line number Diff line
@@ -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')