TensorFlow Классификация

Question

Я использовал TensorFlow для обучения модели, которая могла бы идентифицировать 8 категорий. Я хочу тренироваться на основе исходной модели и увеличивать категорию до 10. Я не хочу тренироваться с нуля.Что мне делать? Спасибо!

Answer 1

Что вы можете сделать, это изменить выходной слой на 10 категорий и установить файл контрольных точек как тот, который вы использовали для 8 категорий.Затем вы продолжаете тренировку с этого момента.Вы можете изучить "трансферное обучение" по этому вопросу.