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

v0.1.13 finish all goals eval (even crash)

Browse Source
This commit is contained in:
wty-yy
2025-12-22 20:37:43 +08:00
parent 989332b951
commit 6bbdceee2f
13 changed files with 94 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
robogauge/utils/helpers.py
View File

@@ -73,6 +73,8 @@ def parse_args():
{"name": "--seed", "type": int, "default": 42, "help": "Random seed."},
{"name": "--write-tensorboard", "action": "store_true", "default": False, "help": "Write tensorboard logs."},
{"name": "--plot-radar", "action": "store_true", "default": False, "help": "Plot radar charts for metrics."},
{"name": "--base-mass", "type": float, "default": 0.0, "help": "Set the base mass of the robot."},
{"name": "--friction", "type": float, "default": 1.0, "help": "Set the ground friction coefficient."},
# Multiprocessing parameters, with different seeds
{"name": "--multi", "action": "store_true", "default": False, "help": "Enable multiprocessing."},

4
robogauge/utils/task_register.py
View File

@@ -65,9 +65,9 @@ class TaskRegister():
sim_cfg.render.save_video = args.save_video
if args.write_tensorboard is not None:
gauger_cfg.write_tensorboard = args.write_tensorboard
if hasattr(args, 'friction') and args.friction is not None:
if args.friction is not None:
sim_cfg.domain_rand.friction = args.friction
if hasattr(args, 'base_mass') and args.base_mass is not None:
if args.base_mass is not None:
sim_cfg.domain_rand.base_mass = args.base_mass
task_register = TaskRegister()