ray.rllib.policy.Policy.导出模型#

Policy.export_model(export_dir: str, onnx: int | None = None) None[源代码]#

将策略的模型导出到本地目录以供服务。

注意:文件格式将取决于所使用的深度学习框架。有关更多详细信息,请参阅Policy的子类及其`export_model`实现。

参数:
  • export_dir – 本地可写目录。

  • onnx – 如果提供,将导出 ONNX 格式的模型。此参数的值设置要使用的 ONNX OpSet 版本。

抛出:

ValueError – 如果一个基于原生DL框架的模型(例如一个keras模型)由于各种原因无法保存到磁盘。