mirror of https://github.com/InternLM/InternLM
feat:add bind numa
parent
cb34b9325b
commit
d9fd76f8a7
|
|
@ -40,7 +40,7 @@ def get_default_parser():
|
||||||
help="launcher for launching distributed environment",
|
help="launcher for launching distributed environment",
|
||||||
)
|
)
|
||||||
parser.add_argument("--host", type=str, help="the master address for distributed training")
|
parser.add_argument("--host", type=str, help="the master address for distributed training")
|
||||||
parser.add_argument("--port", type=int, default=9999, help="the master port for distributed training")
|
parser.add_argument("--port", type=int, default=8888, help="the master port for distributed training")
|
||||||
parser.add_argument("--world_size", type=int, help="world size for distributed training")
|
parser.add_argument("--world_size", type=int, help="world size for distributed training")
|
||||||
parser.add_argument("--rank", type=int, help="rank for the default process group")
|
parser.add_argument("--rank", type=int, help="rank for the default process group")
|
||||||
parser.add_argument("--local_rank", type=int, help="local rank on the node")
|
parser.add_argument("--local_rank", type=int, help="local rank on the node")
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue