ONE - On-device Neural Engine
Loading...
Searching...
No Matches
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 248 of file CheckpointLoader.cc.

251{
252 CheckpointLoader loader(filename);
253 loader.updateTensor(exec);
254 loader.updateOptimizer(train_info, exec);
255 loader.updateTrainingInfo(train_info);
256}