Бета-версия TF 2.0: как повторно инициализировать веса пользовательской модели?

Question

Итак, у меня есть модель, построенная на основе подклассов моделей и слоев.Я делаю некоторые эксперименты, я хотел бы вернуться к недавно инициализированному NN.Как я могу это сделать?В настоящее время я просто запускаю код, который определяет его снова.Там должен быть аккуратный способ сделать это.Я попробовал model.build (запустив его дважды).Не сработало.

Answer 1

Если вы хотите повторно инициализировать с одинаковыми случайными весами, то вы можете просто сохранить веса до тренировки на диск с помощью:

# build model, model.compile(), etc

model.save_weights('initial_weights.h5')

# model.fit(), etc

затем, когда вы захотите вернуться, вы можете снова загрузить начальные веса с помощью:

model.load_weights('initial_weights.h5')