tensorflow关闭迁移学习
时间: 2023-06-12 09:02:14 浏览: 95
您可以在TensorFlow中关闭迁移学习方法,具体方法是在定义模型时将trainable参数设置为False。例如,在定义一个全连接层时,您可以这样写:
```
fc_layer = tf.layers.dense(inputs=input_layer, units=num_units, activation=tf.nn.relu, trainable=False)
```
这将会把这个层设为不可训练,而且不会对该层的权重进行更新。
阅读全文