TypeError: float() argument must be a string or a number, not 'Dimension'

Keras使用报错。

这个也是一个很难搞的bug

不得不说,keras真是垃圾。

原因可能是你使用了  yt = Reshape((1,yt.shape[1]))(yt) 

这句话在tf-cpu上跑是没有问题的。

但如果到了tf-gpu上,就有问题了。

需要改成 yt = Reshape((1,yt.shape[1].value))(yt)

也就是说,利用shape获取维度信息时,应该要使用.value。

这个是github上的讨论。

你可能感兴趣的:(TypeError: float() argument must be a string or a number, not 'Dimension')