Skip to content

Commit

Permalink
update launch.py (#58)
Browse files Browse the repository at this point in the history
update launch.py for supporting the case in which one wanna run multi accelerate instances in single node multigpu. avoiding the port conflict.
  • Loading branch information
Jesse1eung authored Apr 28, 2021
1 parent ae578b2 commit 727e8ee
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions src/accelerate/commands/launch.py
Original file line number Diff line number Diff line change
Expand Up @@ -137,6 +137,7 @@ def multi_gpu_launcher(args):
)
else:
cmd.extend(["--nproc_per_node", str(args.num_processes)])
cmd.extend(["--master_port", str(args.main_process_port)])
cmd.append(args.training_script)
cmd.extend(args.training_script_args)

Expand Down

0 comments on commit 727e8ee

Please sign in to comment.