tensorflow2.0 keras 迁移学习 删除预训练模型的最后一层(layer)
摘要:
Sequential model 方法一、 返回原模型(不包含最后一层)的拷贝 new_model = tf.keras.models.Sequential(base_model.layers[:-1]) 方法二、 原地删除原模型的最后一层 base_model._layers.pop() 函数式 阅读全文
posted @ 2021-06-28 15:51 tianyaDream 阅读(1373) 评论(0) 推荐(0) 编辑