我训练了一个自动编码器,现在想将编码器部分(包括权重)用于其他目的。我可以加载完整模型,获取编码器部分,然后在这些层中设置“可训练”参数吗?或者我是否需要重新编译适应的编码器架构(即所有层设置为不可训练),然后才加载权重?
最佳答案
您可以迭代模型层列表 ( model.layers
) 并修改编码器层,使 trainable
标志为 False
.
for layer in model.layers:
layer.trainable = False
关于python-3.x - 如何加载经过训练的自动编码器(Keras)、获取编码器部分并将其卡住?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/56317547/