类
以下类全局可用。
-
ORT 检查点是模型在给定时间点的状态快照。
此类别包含整个训练会话状态,其中包括模型参数、它们的梯度、优化器参数和用户属性。
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 上查看
类参考