TensorFlow分类

Question

我使用TensorFlow训练了一个可以识别8个类别的模型，我想在原始模型的基础上进行训练，并将类别增加到10个。我不想从头开始训练。我该怎么做?谢谢！

Answer 1

您可以做的是将输出层更改为10个类别，并将检查点文件设置为您用于8个类别的文件。然后你就可以从那时开始继续训练了。关于这一点，你可以研究“迁移学习”。