Python Tensorflow在“之后缓慢”；已完成运行本地“u init”op；

我从github获得了一个基于tensorflow的代码，速度非常慢。它甚至不打印（即使在为tf.logging启用调试模式之后）在打印以下内容之后发生的事情

信息：tensorflow：完成运行本地初始化操作。-->此行代码需要20分钟执行 信息：tensorflow:prediction\u循环标记为已完成

有人能告诉我在哪里查看和优化吗？
已经检查了以下内容：

• model_fn：在本地_init_op之前执行
• 本地文件：从本地文件系统保存和加载检查点文件。因此，文件传输延迟不应该是原因
• 热启动：已尝试。对预测时间没有影响

估算器代码：

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)

---

我认为模型函数已经执行了，根据下面打印的日志信息：tensorflow:name=cls/band/output_bias:0，shape=（2，），INIT_FROM_CKPT----在TPUEstimatorSpec----在TPUEstimatorSpec----信息：tensorflow:Done调用模型。信息：tensorflow：图表已定稿。信息：tensorflow:从/content/ckpt/model.ckpt-10949还原参数信息：tensorflow:运行本地初始化操作。信息：tensorflow:完成运行本地初始化操作。