Как восстановить веса слоев деталей модели с помощью Keras API в TensorFlow 2.0?

Question

У меня есть глубокая модель, и я хочу инициализировать веса слоев деталей модели из другой модели меньшего обученного подмножества. Как я должен сделать?

В тензорном потоке 1.x мы можем использовать tf.train.Saver('List of weights variables') и tf.train.Saver.restore().

Answer 1

Вы можете читать и записывать веса в виде списка массивов NumPy, используя Layer.get_weights() и Layer.set_weights(). Model наследуется от Layer, так что вы можете сделать это с целой моделью или отдельными слоями.

Я бы сделал файловый ввод-вывод, используя обычный код Python (вы используете Python?), Например, некоторые изразличные функции ввода / вывода файлов NumPy https://docs.scipy.org/doc/numpy-1.13.0/reference/routines.io.html. Может быть, есть некоторые методы для этого в TensorFlow, но я надеюсь, что нет.