Checkpoint
Save and restore versioned training state (model + optimizer + epoch).
| Function | Description |
|---|---|
load modelState optimizer dirPath | Validate a Version 2 checkpoint completely, then commit model state, |
| optimizer state, and learning rate in that order. Return the restored epoch. | |
save modelState optimizer epoch dirPath | Save canonical model state, optimizer state, and a Version 2 manifest. |
load
load modelState optimizer dirPathValidate a Version 2 checkpoint completely, then commit model state, optimizer state, and learning rate in that order. Return the restored epoch.
Parameters
modelState:ModelStateoptimizer:IOptimizerdirPath:string
Returns int
save
save modelState optimizer epoch dirPathSave canonical model state, optimizer state, and a Version 2 manifest.
Parameters
modelState:ModelStateoptimizer:IOptimizerepoch:intdirPath:string