v1.0.5; add port cli config
This commit is contained in:
@@ -1,4 +1,7 @@
|
||||
# UPDATE
|
||||
## 20260106
|
||||
### v1.0.5
|
||||
1. `robogauge/scripts/server.py`加入`--port`设置端口配置选项
|
||||
## 20260103
|
||||
### v1.0.4
|
||||
1. 加入save_additional_output参数, 启动会自动记录每个step的latent和weights
|
||||
|
||||
@@ -20,6 +20,7 @@ import uuid
|
||||
from fastapi import FastAPI
|
||||
from pydantic import BaseModel
|
||||
from typing import Dict, Optional
|
||||
import argparse
|
||||
|
||||
from dataclasses import dataclass
|
||||
from robogauge.utils.helpers import parse_args, class_to_dict
|
||||
@@ -92,6 +93,9 @@ def run_api_server(input_queue: multiprocessing.Queue, result_dict: dict, port=9
|
||||
uvicorn.run(app, host="127.0.0.1", port=port, log_level="error")
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument('--port', type=int, default=9973, help='API server port')
|
||||
args_cli = parser.parse_args()
|
||||
print("🤖 RoboGauge Evaluation Server Starting...")
|
||||
ctx = multiprocessing.get_context('spawn')
|
||||
manager = ctx.Manager()
|
||||
@@ -100,7 +104,7 @@ def main():
|
||||
|
||||
api_p = ctx.Process(
|
||||
target=run_api_server,
|
||||
args=(task_queue, results_store),
|
||||
args=(task_queue, results_store, args_cli.port),
|
||||
daemon=True
|
||||
)
|
||||
api_p.start()
|
||||
|
||||
Reference in New Issue
Block a user