在迁移学习中,一般只加载网络的特征提取层,通过Netron可以看到基础网络哪些部分被截取掉了
范例代码:
from tensorflow import keras
# 实例化一个具有预训练权重的基础模型
base_model = keras.applications.Xception(include_top=True)
base_model.save("base_model_with_top.h5")
base_model_without_top = keras.applications.Xception(include_top=False)
base_model_without_top.save("base_model_without_top.h5")
include_top=True vs include_top=False
网友评论