在“完成运行local_init_op”之后，Tensorflow变慢

Question

我有一个基于tensorflow的代码，取自github，非常慢。甚至没有打印（即使在启用了tf.logging的DEBUG模式之后）在打印了以下内容之后

INFO：tensorflow：已运行local_init_op。 ->执行此行代码需要20分钟 INFO：tensorflow：prediction_loop标记为完成

有人可以告诉在哪里进行优化吗？
已经检查了以下内容：

• model_fn：此命令在local_init_op之前执行
• 本地文件：检查点文件是从本地文件系统保存和加载的。因此文件传输延迟不应成为原因
• warm_start_from：已尝试。对预测时间没有影响

估算器代码：

estimator = tf.contrib.tpu.TPUEstimator(
      use_tpu=FLAGS.use_tpu,
      model_fn=model_fn,
      config=run_config,
      warm_start_from = tf.estimator.WarmStartSettings(
            ckpt_to_initialize_from='/content/ckpt',
        ),
      train_batch_size=FLAGS.train_batch_size,
      predict_batch_size=FLAGS.predict_batch_size)

预测代码：

results = estimator.predict(
          predict_input_fn, yield_single_examples=True, checkpoint_path='/content/ckpt/model.ckpt-10949')

执行此代码所需的时间：

results = list(results)