Support binding agent to core

This commit is contained in:
Dominik Moritz Roth 2023-09-02 19:55:35 +02:00
parent ed5081f1b8
commit 695046a123
2 changed files with 7 additions and 0 deletions

View File

@ -21,6 +21,7 @@ scheduler:
repetitions: 3
agents_per_job: 3
reps_per_agent: 1
bind_agent_to_core: True
wandb:
project: '{config[name]}'

View File

@ -294,6 +294,9 @@ class Slate():
Runner = self.runners[runnerName]
if orig_config.consume('scheduler.bind_agent_to_core', False):
os.sched_setaffinity(0, [p_ind % os.cpu_count()])
for r in rep_ids:
self.run_id = r
config = copy.deepcopy(orig_config)
@ -324,6 +327,9 @@ class Slate():
Runner = self.runners[runnerName]
if orig_config.consume('scheduler.bind_agent_to_core', False):
os.sched_setaffinity(0, [p_ind % os.cpu_count()])
with wandb.init(
project=project,
reinit=self.consume(wandbC, 'reinit', DEFAULT_REINIT),