可选的tf.Estimator模型热启动

Question

当前CustomEstimator自动恢复所有变量，或者我可以用WarmStartSettings过滤掉它们。 如果变量在检查点中，如何还原（否则将其初始化）？我正在预训练网络的一部分，并且要训练整个网络，我需要编辑我的热启动设置，因为变量不在检查点中。这是非常麻烦的事情（与pytorch相比）。