Views
No views yet
1import onnxruntime as ort
2import numpy as np
3
4# 加载模型
5session = ort.InferenceSession('burnninghotel/mappo-models/product_manager_model.onnx')
6
7# 准备输入
8state = np.random.randn(1, 128).astype(np.float32)
9global_state = np.random.randn(1, 128).astype(np.float32)
10context = np.random.randn(1, 32).astype(np.float32)
11
12# 运行推理
13inputs = {
14 'state': state,
15 'global_state': global_state,
16 'context': context
17}
18
19outputs = session.run(None, inputs)
20action, log_prob, value = outputspip install onnxruntime