mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-14 20:48:00 +00:00
960e320dff
### Description - Implement Objective-C binding for `ORTTrainingSession` - Add `ORTUtils` utility class to handle conversion between C++ and Objective-C types - Add test case for saving checkpoint - Add unit test cases for `ORTTrainingSession` ### Motivation and Context This PR is part of implementing Objective-C bindings for training API. It implements objective-c binding for training session. The objective-C API closely resembles the C++ API. --------- Co-authored-by: Edward Chen <18449977+edgchen1@users.noreply.github.com>
125 lines
4.4 KiB
Objective-C
125 lines
4.4 KiB
Objective-C
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#ifdef ENABLE_TRAINING_APIS
|
|
#import <Foundation/Foundation.h>
|
|
#include <stdint.h>
|
|
|
|
NS_ASSUME_NONNULL_BEGIN
|
|
|
|
/**
|
|
* An ORT checkpoint is a snapshot of the state of a model at a given point in time.
|
|
*
|
|
* This class holds the entire training session state that includes model parameters,
|
|
* their gradients, optimizer parameters, and user properties. The `ORTTrainingSession` leverages the
|
|
* `ORTCheckpoint` by accessing and updating the contained training state.
|
|
*
|
|
* Available since v1.16.0.
|
|
*
|
|
* @note Note that the training session created with a checkpoint state uses this state to store the entire training
|
|
* state (including model parameters, its gradients, the optimizer states and the properties). The ORTTraingSession
|
|
* does not hold a copy of the checkpoint state. Therefore, it is required that the checkpoint state outlive the
|
|
* lifetime of the training session.
|
|
*/
|
|
@interface ORTCheckpoint : NSObject
|
|
|
|
- (instancetype)init NS_UNAVAILABLE;
|
|
|
|
/**
|
|
* Creates a checkpoint from directory on disk.
|
|
*
|
|
* @param path The path to the checkpoint directory.
|
|
* @param error Optional error information set if an error occurs.
|
|
* @return The instance, or nil if an error occurs.
|
|
*
|
|
* @warning The construction of the checkpoint state requires instantiation of `ORTEnv`.
|
|
* The intialization will fail if the `ORTEnv` is not properly initialized.
|
|
*/
|
|
- (nullable instancetype)initWithPath:(NSString*)path
|
|
error:(NSError**)error NS_DESIGNATED_INITIALIZER;
|
|
|
|
/**
|
|
* Saves a checkpoint to directory on disk.
|
|
*
|
|
* @param path The path to the checkpoint directory.
|
|
* @param includeOptimizerState Flag to indicate whether to save the optimizer state or not.
|
|
* @param error Optional error information set if an error occurs.
|
|
* @return Whether the checkpoint was saved successfully.
|
|
*/
|
|
- (BOOL)saveCheckpointToPath:(NSString*)path
|
|
withOptimizerState:(BOOL)includeOptimizerState
|
|
error:(NSError**)error;
|
|
|
|
/**
|
|
* Adds an int property to this checkpoint.
|
|
*
|
|
* @param name The name of the property.
|
|
* @param value The value of the property.
|
|
* @param error Optional error information set if an error occurs.
|
|
* @return Whether the property was added successfully.
|
|
*/
|
|
- (BOOL)addIntPropertyWithName:(NSString*)name
|
|
value:(int64_t)value
|
|
error:(NSError**)error;
|
|
|
|
/**
|
|
* Adds a float property to this checkpoint.
|
|
*
|
|
* @param name The name of the property.
|
|
* @param value The value of the property.
|
|
* @param error Optional error information set if an error occurs.
|
|
* @return Whether the property was added successfully.
|
|
*/
|
|
- (BOOL)addFloatPropertyWithName:(NSString*)name
|
|
value:(float)value
|
|
error:(NSError**)error;
|
|
|
|
/**
|
|
* Adds a string property to this checkpoint.
|
|
*
|
|
* @param name The name of the property.
|
|
* @param value The value of the property.
|
|
* @param error Optional error information set if an error occurs.
|
|
* @return Whether the property was added successfully.
|
|
*/
|
|
|
|
- (BOOL)addStringPropertyWithName:(NSString*)name
|
|
value:(NSString*)value
|
|
error:(NSError**)error;
|
|
|
|
/**
|
|
* Gets an int property from this checkpoint.
|
|
*
|
|
* @param name The name of the property.
|
|
* @param error Optional error information set if an error occurs.
|
|
* @return The value of the property or 0 if an error occurs.
|
|
*/
|
|
- (int64_t)getIntPropertyWithName:(NSString*)name
|
|
error:(NSError**)error __attribute__((swift_error(nonnull_error)));
|
|
|
|
/**
|
|
* Gets a float property from this checkpoint.
|
|
*
|
|
* @param name The name of the property.
|
|
* @param error Optional error information set if an error occurs.
|
|
* @return The value of the property or 0.0f if an error occurs.
|
|
*/
|
|
- (float)getFloatPropertyWithName:(NSString*)name
|
|
error:(NSError**)error __attribute__((swift_error(nonnull_error)));
|
|
|
|
/**
|
|
*
|
|
* Gets a string property from this checkpoint.
|
|
*
|
|
* @param name The name of the property.
|
|
* @param error Optional error information set if an error occurs.
|
|
* @return The value of the property.
|
|
*/
|
|
- (nullable NSString*)getStringPropertyWithName:(NSString*)name
|
|
error:(NSError**)error __attribute__((swift_error(nonnull_error)));
|
|
|
|
@end
|
|
|
|
NS_ASSUME_NONNULL_END
|
|
|
|
#endif // ENABLE_TRAINING_APIS
|