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