Function getProgressBarCallbackConfig

mnist-transfer-cnn/ui.js:67–97

(epochs)

}

export function getProgressBarCallbackConfig(epochs) {
  // Custom callback for updating the progress bar at the end of epochs.

  const trainProg = document.getElementById('trainProg');
  let beginMillis;
  const progressBarCallbackConfig = {
    onTrainBegin: async (logs) => {
      beginMillis = tf.util.now();
      status(
          'Please wait and do NOT click anything while the model retrains...',
          'blue');
      trainProg.value = 0;
    },
    onTrainEnd: async (logs) => {
      // allow retraining again
      document.getElementById('retrain').disabled = false;
      status(
          `Done retraining ${epochs} epochs (elapsed: ` +
              `${(tf.util.now() - beginMillis).toFixed(1)} ms` +
              `). Standing by.`,
          'black');
    },
    onEpochEnd: async (epoch, logs) => {
      status(
          `Please wait and do NOT click anything while the model ` +
          `retrains... (Epoch ${epoch + 1} of ${epochs})`);
      trainProg.value = (epoch + 1) / epochs * 100;
    },
  };
  return progressBarCallbackConfig;
}

export function setPredictError(text) {
  const predictHeader = document.getElementById('predict-header');