Skip to content

Commit 1060e65

Browse files
resolve _generate_run_script_train conflict
1 parent 5352a11 commit 1060e65

1 file changed

Lines changed: 15 additions & 2 deletions

File tree

flagscale/runner/runner_train.py

Lines changed: 15 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -175,7 +175,14 @@ def _get_runner_cmd_train(
175175

176176

177177
def _generate_run_script_train(
178-
config, host, node_rank, cmd, background=True, with_test=False, enable_monitoring=True
178+
config,
179+
host,
180+
node_rank,
181+
cmd,
182+
background=True,
183+
with_test=False,
184+
root_dir=None,
185+
enable_monitoring=False,
179186
):
180187
system_config = config.train.system
181188
logging_config = config.train.system.logging
@@ -787,7 +794,13 @@ def _run_each(
787794
cmd = shlex.join(export_cmd + runner_cmd + [self.user_script] + self.user_args)
788795

789796
host_run_script_file = _generate_run_script_train(
790-
self.config, host, node_rank, cmd, background=False, with_test=with_test
797+
self.config,
798+
host,
799+
node_rank,
800+
cmd,
801+
background=False,
802+
with_test=with_test,
803+
enable_monitoring=enable_monitoring,
791804
)
792805

793806
run_local_command(f"bash {host_run_script_file}", dryrun)

0 commit comments

Comments
 (0)