Commit c147a3e8 authored by Kazuhiro Yamasaki's avatar Kazuhiro Yamasaki
Browse files

Change distributed argument group title to appropriate one.

parent 19301985

megatron/arguments.py

+1 −1
Original line number Diff line number Diff line
@@ -341,7 +341,7 @@ def _add_mixed_precision_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,

                       help='Size of the model parallel.')