Interface TrainingSessionCreateOptions

Represents the optional parameters that can be passed into the TrainingSessionFactory.

interface TrainingSessionCreateOptions {
    checkpointState: UriOrBuffer;
    evalModel?: UriOrBuffer;
    optimizerModel?: UriOrBuffer;
    trainModel: UriOrBuffer;
}

Properties

checkpointState: UriOrBuffer

URI or buffer for a .ckpt file that contains the checkpoint for the training model.

evalModel?: UriOrBuffer

Optional. URI or buffer for the .onnx eval model file.

optimizerModel?: UriOrBuffer

Optional. URI or buffer for the .onnx optimizer model file.

trainModel: UriOrBuffer

URI or buffer for the .onnx training file.