Glossary#

TrainingAgent#

This classes creates a training graph assuming the gradient is computed outside ONNX. See training_agent.cc.

TrainingSession#

Extension of InferenceSession to train a model ONNX. It assumes the input graph contains nodes computing a loss. The gradient is computed based on that. See training_session.cc.

OrtModuleGraphBuilderConfiguration#

See ortmodule_graph_builder.cc.