Сохранение и повторное использование моделей CNN - сохранение инициализаций - PullRequest
0 голосов
/ 21 мая 2019

Я хочу повторить серию экспериментов по классификации изображений, повторно используя CNN с тем же CNN с идентичными гиперпараметрами, особенно при инициализации. Итак, если я сохраню модель после того, как создаю ее экземпляр и до того, как обучу ее, сохраняет ли это также инициализации, чтобы потом потом перезагрузить ее и обучить с другим набором данных и метками, запускает ли она новую модель с теми же гиперпараметрами и инициализации в качестве первой модели, которую я обучил с использованием первого набора данных / классификационных меток? В настоящее время я использую fastai, который, конечно, является библиотекой / набором API, построенных на Pythorch, но я думаю, что всем поможет более общее объяснение, которое охватывает все CNN, использующие любую библиотеку.

Я ожидаю ответа, который говорит: «После этой точки в рабочем процессе, создающем CNN, модель инициализируется, и если вы сохраните ее в этой точке, вы можете перезагрузить ее позже и использовать те же гиперпараметры и инициализации в вашей следующей модели . "

1 Ответ

1 голос
/ 27 мая 2019

вы можете сохранить ученика, как только он будет создан.

Пример:

learn = cnn_learner(data,models.resnet34,metrics=error_rate)
learn.save('init')

позже:

learn.load('init)
...