использовать существующую модель, но требуется больше меток - PullRequest
0 голосов
/ 11 января 2020

У меня есть предварительно обученная модель изображения, которая имеет 4 метки. В новой модели мне бы хотелось иметь две из четырех меток и мне нужно добавить еще одну новую метку. У меня есть около 100 наборов данных, которые можно использовать для обучения модели для нового ярлыка, и я буду использовать имеющиеся знания из предварительно обученной модели для этих двух существующих ярлыков. Есть ли способ, которым я могу обучить эту новую модель с набором данных 100 на основе предварительно обученной модели? Спасибо

1 Ответ

1 голос
/ 11 января 2020

вы можете просто обучить свой последний слой полностью подключен для обучения вашего нового лейбла

...