Tensorflow Saver能否自适应地恢复模型?

时间:2018-04-01 08:24:42

标签: tensorflow

我已经使用tf.train.Saver保存了TF模型,下次我将修改原始模型,例如添加新图层或保留一些图层无法处理。但是当我恢复旧模型时,我收到一个错误,即在checkpoint中找不到某个密钥。有人告诉我,我可以维护var_list来存储我想要恢复的变量并部分恢复模型。但实际上这并不容易,因为有很多变量,包括权重,偏差和来自优化器或批量规范化的一些变量。有没有办法让它更方便?或者TF可以通过判断是否可以在模型文件中找到变量的名称来部分自动恢复模型?如果找到这样的密钥,它只能初始化变量。

0 个答案:

没有答案