ONE - On-device Neural Engine
All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
onert::loader::train Namespace Reference

Functions

void loadCheckpoint (const std::string &filename, const std::unique_ptr< ir::train::TrainingInfo > &train_info, const std::unique_ptr< exec::Execution > &exec)
 

Function Documentation

◆ loadCheckpoint()

void onert::loader::train::loadCheckpoint ( const std::string &  filename,
const std::unique_ptr< ir::train::TrainingInfo > &  train_info,
const std::unique_ptr< exec::Execution > &  exec 
)

Definition at line 244 of file CheckpointLoader.cc.

247{
248 CheckpointLoader loader(filename);
249 loader.updateTensor(exec);
250 loader.updateOptimizer(train_info, exec);
251 loader.updateTrainingInfo(train_info);
252}