我训练了自动编码器,现在想将编码器零件(包括砝码)用于其他目的。我可以加载完整的模型,获取编码器零件,然后在这些层中设置“可训练”参数吗?还是我需要重新编译适应的编码器架构(即所有层均设置为不可训练),然后才加载权重?

最佳答案

您可以遍历图层的模型列表(model.layers)并修改编码器的图层,以使trainable标志为False

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

关于python-3.x - 如何加载训练有素的自动编码器(Keras),使用编码器零件并将其卡住?,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/56317547/

10-12 22:07