类
以下类可在全局范围内使用。
-
ORT 检查点(Checkpoint)是模型在特定时间点的状态快照。
该类保存了整个训练会话的状态,包括模型参数、它们的梯度、优化器参数和用户属性。
ORTTrainingSession通过访问和更新其中包含的训练状态来利用ORTCheckpoint。从 1.16 版本开始提供。
查看更多注意
此类仅在启用训练 API 时可用。声明
Objective-C
@interface ORTCheckpoint : NSObjectSwift
class ORTCheckpoint : NSObject -
用于配置 CoreML 执行提供程序的选项。
查看更多声明
Objective-C
@interface ORTCoreMLExecutionProviderOptions : NSObjectSwift
class ORTCoreMLExecutionProviderOptions : NSObject -
用于配置会话的选项。
查看更多声明
Objective-C
@interface ORTSessionOptions : NSObjectSwift
class ORTSessionOptions : NSObject -
声明
Objective-C
@interface ORTEnv : NSObjectSwift
class ORTEnv : NSObject -
ORT 会话用于加载和运行模型。
查看更多声明
Objective-C
@interface ORTSession : NSObjectSwift
class ORTSession : NSObject -
用于配置运行的选项。
查看更多声明
Objective-C
@interface ORTRunOptions : NSObjectSwift
class ORTRunOptions : NSObject -
提供训练、评估和优化 ONNX 模型方法的训练器类。
训练会话需要四个训练构件
- 训练 ONNX 模型
- 评估 ONNX 模型(可选)
- 优化 ONNX 模型
- 检查点目录
onnxruntime-training python 工具 可用于生成上述训练制品。
从 1.16 版本开始提供。
查看更多注意
此类仅在启用训练 API 时可用。声明
Objective-C
@interface ORTTrainingSession : NSObjectSwift
class ORTTrainingSession : NSObject -
ORT 值封装了运行时作为模型输入或输出使用的数据。
查看更多声明
Objective-C
@interface ORTValue : NSObjectSwift
class ORTValue : NSObject -
值的类型信息。
查看更多声明
Objective-C
@interface ORTValueTypeInfo : NSObjectSwift
class ORTValueTypeInfo : NSObject -
张量的类型和形状信息。
查看更多声明
Objective-C
@interface ORTTensorTypeAndShapeInfo : NSObjectSwift
class ORTTensorTypeAndShapeInfo : NSObject -
用于配置 Xnnpack 执行提供程序的选项。
查看更多声明
Objective-C
@interface ORTXnnpackExecutionProviderOptions : NSObjectSwift
class ORTXnnpackExecutionProviderOptions : NSObject
在 GitHub 上查看
类参考