using whileTraining callback always when specified

ml5js · Oct 29, 2020 · bcca5c5 · bcca5c5
1 parent 93e6024
commit bcca5c5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/NeuralNetwork/index.js b/src/NeuralNetwork/index.js
@@ -515,7 +515,7 @@ class DiyNeuralNetwork {
  options.whileTraining = [
  this.neuralNetworkVis.trainingVis(),
  {
- onEpochEnd: null,
+ onEpochEnd: whileTrainingCb,
  },
  ];
  } else {