在Tensorflow

时间:2017-04-07 13:30:14

标签: python tensorflow

我有一个预训练的inception_v3模型,我使用这个函数来加载冻结的权重

tf.import_graph_def(graph_def)

对于我的任务,我还想更改/重置此inception_v3模型的权重。

我可以通过

获取所有变量
op = sess.graph.get_operations()

另一方面,我只想要那些具有权重的变量。在这种情况下,tf.trainable_variables()会返回None

tl; dr:我想从冻结图中提取权重和偏差变量

0 个答案:

没有答案