我想做转移学习,我已经加载这些权重文件,但现在我迷失在如何使用它的层来训练我的自定义模型。 任何帮助都将不胜感激 下面是我尝试的示例代码:
local_weights_file= '/tmp/inception_v3_weights_tf_dim_ordering_tf_kernels_notop.h5'
pre_trained_model = InceptionV3(input_shape = (150, 150, 3),
include_top = False,
weights = None)
pre_trained_model.load_weights(local_weights_file)
for layer in pre_trained_model.layers:
layer.trainable = False
您需要将最后一层的输出带到,并将其输入到最终模型中。 像这样的东西应该管用
相关问题 更多 >
编程相关推荐