标签: python tensorflow
我有一个预训练的inception_v3模型,我使用这个函数来加载冻结的权重
tf.import_graph_def(graph_def)
对于我的任务,我还想更改/重置此inception_v3模型的权重。
我可以通过
op = sess.graph.get_operations()
另一方面,我只想要那些具有权重的变量。在这种情况下,tf.trainable_variables()会返回None。
tf.trainable_variables()
None
tl; dr:我想从冻结图中提取权重和偏差变量