如何从``tensorflow_hub''(在Tensorflow 2.0中)仅微调模型的某些层?

时间:2019-11-18 16:14:54

标签: tensorflow tensorflow2.0

假设我有一个这样创建的模型(请参见here):

base_model = tf.keras.applications.MobileNetV2(...)

然后我只能冻结一些图层:

base_model.trainable = True
for layer in base_model.layers[:fine_tune_at]:
    layer.trainable =  False

现在假设我创建了一个像这样的模型:

hub_model = hub.KerasLayer(hub_url,
                           ...
                           # parameters are unfreezed
                           trainable=True)

如何仅微调此模型的某些层?还是不支持?

0 个答案:

没有答案