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

v0.1.16; Almost finish StressPipeline

Browse Source
This commit is contained in:
wty-yy
2025-12-26 23:51:43 +08:00
parent 65490fe591
commit dad65a4e27
11 changed files with 125 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
robogauge/scripts/run.py
View File

@@ -13,8 +13,7 @@ os.environ["OMP_NUM_THREADS"] = "1"
os.environ["MKL_NUM_THREADS"] = "1"
from robogauge.tasks import *
from robogauge.tasks.pipeline.multi_pipeline import MultiPipeline
from robogauge.tasks.pipeline.level_pipeline import LevelPipeline
from robogauge.tasks.pipeline import *
from robogauge.utils.task_register import task_register
from robogauge.utils.helpers import parse_args
@@ -23,7 +22,10 @@ from robogauge.utils.logger import logger
if __name__ == '__main__':
args = parse_args()
if args.multi:
if args.stress_benchmark:
stress_pipeline = StressPipeline(args)
stress_pipeline.run()
elif args.multi:
multi_pipeline = MultiPipeline(args)
multi_pipeline.run()
elif args.search_max_level: