【tensorflow】fine-tuning, 选择性加载ckpt部分权重

在上一篇的转载博客中http://blog.csdn.net/shwan_ma/article/details/78874696中，

我发现用slim可以很方便的选择性的从ckpt文件中加载所要的数据。

以我自己的代码为例：

exclude = ['layer_weight','WD2_conv', 'BD2']
variables_to_restore=slim.get_variables_to_restore(exclude=exclude)
self.saver = tf.train.Saver(variables_to_restore)
self.saver.restore(self.sess, filename)

上述代码指的是除了权重名称为“layer_weight”，”WD2_conv”， “BD2”三种权重外，加载其他的权重。当然我不加载这些权重是为了更好的fine-tuning。

注意：

这是我犯的一个小错误，就是在保存重新训练的