TPU过时的所有张量都应保留批次大小尺寸,但标量张量

时间:2019-10-28 11:37:53

标签: python tensorflow tpu

我有一个我不明白的张量流错误。提示问题可能在哪里会对我有很大帮助。

错误如下:

  

在create_tpu_hostcall中,文件“ /usr/local/lib/python3.6/dist-packages/tensorflow_estimator/python/estimator/tpu/tpu_estimator.py”,第2214行       '尺寸,但标量为{}'。format(dequeue_ops [i] [0]))

     

RuntimeError:从TPU输出的所有张量都应保留批大小尺寸,但标量Tensor(“ OutfeedDequeueTuple:0”,shape =(),dtype = float32,device = / job:worker / task:0 / device:CPU :0)

代码在google colab中运行。我尝试使用XLNet运行多标签分类,因此我在这里和那里修改了代码。 评估时发生错误,训练完成但没有错误:

    for global_step, filename in sorted(steps_and_files, key=lambda x: x[0]):
      ret = estimator.evaluate(
      input_fn=eval_input_fn,
      steps=eval_steps,
      checkpoint_path=filename)

该错误是否会被某人敲响?

0 个答案:

没有答案