restoreTrainingCheckpoint function

void restoreTrainingCheckpoint(TrainingD training, Map<String, dynamic> checkpoint)

Restores a checkpoint produced by saveTrainingCheckpoint into training (which must have the same topology/algorithm/hyperparameters).

Implementation

void restoreTrainingCheckpoint(
  TrainingD training,
  Map<String, dynamic> checkpoint,
) {
  final w = (checkpoint['annWeights'] as List)
      .map((e) => (e as num).toDouble())
      .toList();
  training.ann.allWeights = w;
  if (training is GradientOptimizer && checkpoint['optimizerState'] != null) {
    final opt = training as GradientOptimizer;
    opt.optimizerStep = (checkpoint['optimizerStep'] as num).toInt();
    opt.loadOptimizerState(checkpoint['optimizerState'] as List);
    if (checkpoint['gradients'] != null) {
      opt.loadGradients(checkpoint['gradients'] as List);
    }
    if (checkpoint['globalLearnError'] != null) {
      opt.restoreGlobalLearnErrors(
        (checkpoint['lastGlobalLearnError'] as num).toDouble(),
        (checkpoint['globalLearnError'] as num).toDouble(),
      );
    }
  }
}