from local

This commit is contained in:
dohe0342 2023-01-21 15:34:51 +09:00
parent 5b23ae081e
commit a52d8af88d
2 changed files with 6 additions and 0 deletions

View File

@ -96,6 +96,12 @@ def get_parser():
parser = argparse.ArgumentParser(
formatter_class=argparse.ArgumentDefaultsHelpFormatter
)
parser.add_argument(
"--group-num",
type=int,
default=0,
help="Number of groups",
)
parser.add_argument(
"--world-size",