Сообщается как проблема в https://github.com/tensorflow/tfjs/issues/2727
Казалось, что использование опции initialEpochs для подгонки привело к завершению подгонки без каких-либо тренировок. Тогда я обнаружил, что это было потому, что initialEpochs был равен эпохам. Так, например,
{initialEpochs: 5, epochs: 8}
будет обучать 3 эпохи. Может быть, кто-то должен исправить документацию, которая гласит:
epochs (число) Число раз, которое нужно перебрать по массивам обучающих данных.
Это не делает еще 8 эпохи, начинающиеся с 5. Документация должна сказать что-то о эпохах, определяющих, на каком номере эпохи останавливаться.
Кроме того, tf js -vis всегда dr aws графы, где первая эпоха помечена 0, а не initialEpoch. Обратите внимание, что обратные вызовы, такие как onEpochEnd, получают правильный номер эпохи.