删除TF2训练模型的最后一层并保存的方法

时间:2020-09-15 07:52:13

标签: tensorflow2.0 pre-trained-model

我指的是一些链接,以删除模型的最后一层并保存以供将来预测。

How to remove the last layer from trained model in Tensorflow

只是想知道是否还有更多方法可以做到这一点。例如

from tensorflow.keras import layers, losses
from tensorflow.keras.models import Model
model = tf.keras.Sequential([
    layers.Dense(64,activation='relu',input_shape=(None,784)),
    layers.Dense(128, activation='relu'), 
    layers.Dense(10,activation='sigmoid')
])
model.compile(optimizer='adam', loss=losses.SparseCategoricalCrossentropy())


model1 = Model(inputs = model.input, outputs= model.layers[-2].output)
model1.save()

model1是新模型。

我的假设是新模型将提供经过训练的权重。如果我错了,请纠正我。

0 个答案:

没有答案