Как продолжить тренировать модель с новыми классами и данными? - PullRequest
0 голосов
/ 12 мая 2018

Я успешно обучил модель и теперь хочу продолжить обучение с новыми данными. Если данные даны с одинаковым количеством классов, это работает нормально. Но имея больше данных, чем первоначально, я получу ошибку:

ValueError: фигуры (?, 14) и (?, 21) несовместимы

Как я могу динамически увеличить количество классов в моей обученной модели или как заставить модель принимать меньшее количество классов? Нужно ли сохранять классы в файле рассола?

1 Ответ

0 голосов
/ 12 мая 2018
  • Лучше всего обучить вашу сеть с нуля, настроив выходные слои в соответствии с новым размером выходного класса.
  • Если переподготовка является проблемой, то оставьте обученную сеть такой, какая она есть, и отбросьте только последний уровень. Добавьте новый слой с правильным выходным размером, инициализируйте его случайными весами, а затем настройте (обучите) всю сеть.
...