更改预训练模型的辍学率以进行推理

时间:2020-03-12 09:31:55

标签: python tensorflow keras

我使用Keras和Tensorflow 2.0,并希望更改预训练模型的辍学率。我尝试了以下无效的方法:

model_path = './model_1.h5'
model = tf.keras.models.load_model(model_path)
f = K.function([model.inputs, K.learning_phase(). K.dropout(level=0.9)], [model.outputs])

有没有简单的方法可以做到这一点?

0 个答案:

没有答案