可选热启动tf.估计器国防部

2024-06-16 10:50:09 发布

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

当前CustomEstimator自动恢复所有变量,或者我可以用WarmStartSettings过滤掉它们。 如果变量在检查点中,如何还原它们(否则初始化它们)?我正在预先训练网络的一部分,为了训练完整的网络,我需要编辑我的热启动设置,因为变量不在检查点中。这是非常笨重的(与Pythorch相比)。在


Tags: 网络编辑pythorchwarmstartsettingscustomestimator