我有一个我不明白的张量流错误。提示问题可能在哪里会对我有很大帮助。
错误如下:
在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)
该错误是否会被某人敲响?