Как я могу загрузить обученный автоэнкодер (Keras), взять часть энкодера и заморозить его? - PullRequest
2 голосов
/ 27 мая 2019

Я обучил автоэнкодеру и теперь хотел бы использовать часть энкодера (включая веса) для другой цели.Могу ли я загрузить полную модель, получить часть датчика и затем установить параметр «trainable» в этих слоях?Или мне нужно перекомпилировать адаптированную архитектуру кодировщика (т. Е. Все уровни, для которых установлен ненарушимый), и только затем загружать веса?

1 Ответ

0 голосов
/ 02 июня 2019

Вы можете выполнить итерацию по списку моделей слоев (model.layers) и изменить слои кодировщика так, чтобы флаг trainable был False.

for layer in model.layers:
    layer.trainable = False
...