Как сохранить две лучшие модели для набора различных данных в pytorch - PullRequest
0 голосов
/ 25 апреля 2019

У меня есть модель глубокого обучения Pytorch.После каждой тренировочной эпохи я проверяю точность для набора проверки и, если он больше, чем последняя лучшая точность, я сохраняю соответствующую модель и контрольные точки.Теперь я хочу использовать ту же модель, в которой уже обучен простой набор данных, и настроить его на более сложный набор данных (возобновив обучение на втором наборе данных).очевидно, я не достигаю той же хорошей точности, которую получил от простого набора данных, но я получаю лучшую точность, чем только тренировка на сложном наборе данных (без предварительной подготовки на простом наборе данных).Моя проблема заключается в том, как управлять и сохранить лучшую модель для второго набора данных (сложного).

Я ценю ваше руководство.

...