2021年6月28日

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 阅读(1344) 评论(0) 推荐(0) 编辑

导航