保存估算器时忽略某些变量

时间:2019-06-07 11:39:02

标签: python tensorflow tensorflow-estimator

我有一个自定义估算器,它使用vgg模型进行特征匹配(我需要在vgg激活上进行渐变)-因此,vgg需要成为我模型的一部分(model_fn)。这些vgg变量未经过优化,并使用自定义功能进行了还原(vgg变量是从单独的目录分配的,因此所有模型都包含一个变量)。 我不想在检查点中加载或保存这些变量。如何使用tf.estimator.Estimator进行操作? 我知道我可以使用WarmStartSettings过滤变量,但这很麻烦,并且变量仍然保存。

有类似的问题: How to remove untrainable variables when saving checkpoint with tensorflow estimator? 但是我找不到改变保存程序的方法(看起来像那个人手动保存了模型吗?)。

0 个答案:

没有答案