gpt4 book ai didi

tensorflow - 使用 tf.estimator.Estimator 加载检查点和微调

转载 作者:行者123 更新时间:2023-12-03 00:36:56 27 4
gpt4 key购买 nike

我们正在尝试将旧的训练代码转换为更符合 tf.estimator.Estimator 的代码。在初始代码中,我们针对目标数据集微调原始模型。在使用 variables_to_restoreinit_fnMonitoredTrainingSession 的组合进行训练之前,仅从检查点加载一些层。如何使用 tf.estimator.Estimator 方法实现这种重量加载?

最佳答案

您有两个选择,第一个更简单:

1- 在 model_fn 中使用 tf.train.init_from_checkpoint

2- model_fn 返回一个 EstimatorSpec。您可以通过EstimatorSpec设置脚手架。

关于tensorflow - 使用 tf.estimator.Estimator 加载检查点和微调,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/46423956/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com