class

OnnxRuntime::TrainingSession

Inherits Reference / Object

TrainingSession class provides high-level API for training ONNX models. This is a placeholder for future implementation when ONNX Runtime training API is supported.

Constructors

new(env : OrtEnvironment, model_path : String, **options)

Creates a new TrainingSession instance.

Source

Instance methods

eval_step(input_feed)

Evaluate the model

Source

learning_rate=(rate : Float64)

Set learning rate

Source

load_checkpoint(checkpoint_path : String)

Load a checkpoint

Source

optimizer_state

Get optimizer state

Source

save_checkpoint(checkpoint_path : String)

Save the trained model

Source

train_step(input_feed)

Train the model for one step

Source