Skip to content

Commit

Permalink
modify argument help sentence
Browse files Browse the repository at this point in the history
  • Loading branch information
eunwoosh committed Dec 13, 2022
1 parent 93666f9 commit a2d5e2d
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions otx/cli/tools/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -140,7 +140,7 @@ def parse_args():
parser.add_argument(
"--gpus",
type=str,
help="Enable multi gpu training. if number of gpu is more than one, then model is trained in each gpu with splited dataset.",
help="Comma-separated indcies of GPU. If there are more than one available GPU, then model is trained with multi GPUs.",
)
parser.add_argument(
"--multi-gpu-port",
Expand Down Expand Up @@ -322,8 +322,9 @@ def run_child_process(rank: int, gpu_ids: List[int], output_path: str, multi_gpu
if "--enable-hpo" in sys.argv:
sys.argv.remove('--enable-hpo')
MultiGPUManager.set_arguments_to_argv("--save-logs-to", output_path)

MultiGPUManager.initialize_multigpu_train(rank, gpu_ids, multi_gpu_port)
print("*"*100, os.getpid())

main()

@staticmethod
Expand Down

0 comments on commit a2d5e2d

Please sign in to comment.