I trained an autoencoder, and would now like to use the encoder part (including weights) for another purpose. Can I load the full model, get the encoder part, and then set the 'trainable' parameter in these layers? Or do I need to recompile the adapted encoder architecture (i.e. all layers set to untrainable), and only then load the weights?
You can iterate through the model list of layers ( model.layers
) and modify the layers of encoder so that trainable
flag is False
.
for layer in model.layers:
layer.trainable = False