cyy/RoboGauge
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

v1.0.3; fix multiprocess global variables bug

Browse Source
This commit is contained in:
wty-yy
2026-01-02 11:06:06 +08:00
parent 82683ba394
commit c81bdec934
4 changed files with 69 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
robogauge/utils/task_register.py
View File

@@ -33,9 +33,9 @@ class TaskRegister():
def get_cfgs(self, name):
if name not in self.sim_cfgs:
raise ValueError(f"Task '{name}' is not registered, checkout '{ROBOGAUGE_ROOT_DIR}/robogauge/tasks/__init__.py'.")
sim_cfg = self.sim_cfgs[name]
gauger_cfg = self.gauger_cfgs[name]
robot_cfg = self.robot_cfgs[name]
sim_cfg = self.sim_cfgs[name]()
gauger_cfg = self.gauger_cfgs[name]()
robot_cfg = self.robot_cfgs[name]()
return sim_cfg, gauger_cfg, robot_cfg
def make_pipeline(self, args=None, sim_cfg=None, gauger_cfg=None, robot_cfg=None, create_logger=True):
@@ -55,7 +55,7 @@ class TaskRegister():
run_name = args.run_name + f'_{args.seed}'
if create_logger:
logger.create(args.experiment_name, run_name)
return pipeline_class(run_name, sim_cfg, robot_cfg, gauger_cfg)
return pipeline_class(run_name, sim_cfg, robot_cfg, gauger_cfg, args)
def update_args_to_cfg(self, sim_cfg, gauger_cfg, robot_cfg, args):
if args.model_path is not None: