如何在tf.get_collection中创建一个条件

时间:2017-04-28 11:55:35

标签: python-3.x tensorflow

我有一个ckpt,我想用tf.train.Saver()恢复,我想只恢复所有层,因为我想使用最低层进行传输学习这里是我的代码

reuse_vars = tf.get_collection(tf.GraphKeys.TRAINABLE_VARIABLES, scope = "")
reuse_vars_dict = dict([(var.name, var) for var in original_saver = 
tf.train.Saver(reuse_vars_dict)

以下是我为图层命名的方法

48-convolutional/biases:0
48-convolutional/kernel:0
48-convolutional/gamma:0
50-convolutional/biases:0
50-convolutional/kernel:0

如何确定范围不对50层元数据收费

我使用Tensorflow

我需要帮助

谢谢

0 个答案:

没有答案