Function setRetrainFunction

mnist-transfer-cnn/ui.js:59–65

(retrain)

}

function setRetrainFunction(retrain) {
  const retrainButton = document.getElementById('retrain');
  retrainButton.addEventListener('click', async () => {
    document.getElementById('retrain').disabled = true;
    await retrain();
  });
}

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