关于tf.keras的模型加载

加载模型使用model.load_weights函数

model.load_weights(model_path, by_name=True, skip_mismatch=True)

by_name=False 的时候按照网络的拓扑结构加载权重

by_name=True 的时候就是按照网络层名称进行加载

skip_mismatch:bool值, 这个只有在by_name=True时才会生效, 当具有相同名字的层的权重个数或形状不一样时,就不加载这些权重.

你可能感兴趣的:(tensorflow踩坑日记,keras,深度学习,人工智能)