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上的讨论

发布了449 篇原创文章 · 获赞 197 · 访问量 25万+

猜你喜欢

转载自blog.csdn.net/qq_40774175/article/details/105196387