在TPU上运行Keras应用程序

时间:2019-03-07 13:29:53

标签: python tensorflow keras tpu tf.keras

我目前正在尝试在TPU上运行预制的Keras模型。我已经使模型接近正常工作,但是遇到了无法解决的错误:

  

TypeError:提要的值不能是tf.Tensor对象。可接受的提要值包括Python标量,字符串,列表,numpy ndarray或TensorHandles。作为参考,张量对象是Tensor(“ strided_slice_3:0”,shape =(7,1),dtype = int64),该参数已传递给提要带有键Tensor(“ infeed-enqueue-dense_target_30-1:0”,shape =(7,1),dtype = float32,device = / job:worker / device:TPU:1)。

由于相同的问题,我无法找到其他人,因此我尝试了一些修复。我尝试使用其他应用程序,例如ResNet50,InceptionV3和NASNetLarge,但在不同张量下都失败了。我还尝试在v2和v3 TPU上运行代码,但都失败了。

我必须使用Tensorflow r1.12,因为TPU不支持r1.13。

我的代码在https://github.com/ehiggins98/X-Ray-Analysis

0 个答案:

没有答案