TensorFlow2.1.0报错解决:ValueError: Failed to convert a NumPy array to a Tensor (Unsupported object type

Keras训练模型报错。


运行环境:

  • cudatoolkit = 10.1.243
  • cudnn = 7.6.5
  • tensorflow-gpu = 2.1.0

报错:

ValueError: Failed to convert a NumPy array to a Tensor (Unsupported object type float)


原因:
fit()之前没有把训练样本和测试样本转化成Keras可接受的数据类型。


解决:

fit() 之前先转换数据类型:

trainX = trainX.astype('float64')
testX = testX.astype('float64')

你可能感兴趣的:(TensorFlow,&,Keras)