保存estim时忽略某些变量

2024-06-16 09:41:36 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个定制的估计器,它使用vgg模型进行特征匹配(我需要vgg激活的梯度)-因此vgg需要成为我模型的一部分(model\u fn)。这些vgg变量没有经过优化,而是使用自定义函数还原的(vgg变量是从单独的目录中分配的,所以所有模型都有一个)。 我不想在我的检查点中加载或保存这些变量。我怎样才能用tf.estimator.estimator做到这一点? 我知道我可以用WarmStartSettings过滤我的变量,但它是cumberstone,变量仍然保存

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


Tags: to函数模型目录modeltf特征remove