TensorFlow 分类

Question

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

Answer 1

你可以将输出层改为10个类别，并将检查点文件设置为你用于8个类别的那个。然后你从那一刻开始继续训练。你可以研究一下这方面的“迁移学习”。