Skip to content

GitLab

Commit 356f8771 authored by Jared Casper's avatar Jared Casper
Browse files

Merge branch 'fix/help-title-dist' of https://github.com/lazykyama/Megatron-LM into community-fixes

parents 24ccd8aa c147a3e8
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
megatron/arguments.py
View file @ 356f8771
...@@ -358,7 +358,7 @@ def _add_mixed_precision_args(parser): ...@@ -358,7 +358,7 @@ def _add_mixed_precision_args(parser):
def _add_distributed_args(parser): def _add_distributed_args(parser):
group = parser.add_argument_group(title='mixed precision') group = parser.add_argument_group(title='distributed')
group.add_argument('--model-parallel-size', type=int, default=1, group.add_argument('--model-parallel-size', type=int, default=1,
help='Size of the model parallel.') help='Size of the model parallel.')
......
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