网络编程
位置:首页>> 网络编程>> Python编程>> 解决Keras TensorFlow 混编中 trainable=False设置无效问题

解决Keras TensorFlow 混编中 trainable=False设置无效问题

作者:芥末的无奈  发布时间:2022-10-23 16:56:19 

标签:Keras,TensorFlow,trainable,False

这是最近碰到一个问题,先描述下问题:

首先我有一个训练好的模型(例如vgg16),我要对这个模型进行一些改变,例如添加一层全连接层,用于种种原因,我只能用TensorFlow来进行模型优化,tf的优化器,默认情况下对所有tf.trainable_variables()进行权值更新,问题就出在这,明明将vgg16的模型设置为trainable=False,但是tf的优化器仍然对vgg16做权值更新

来源:https://blog.csdn.net/weiwei9363/article/details/79673201

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com