Keras使用报错。
这个也是一个很难搞的bug
不得不说,keras真是垃圾。
原因可能是你使用了 yt = Reshape((1,yt.shape[1]))(yt)
这句话在tf-cpu上跑是没有问题的。
但如果到了tf-gpu上,就有问题了。
需要改成 yt = Reshape((1,yt.shape[1].value))(yt)
也就是说,利用shape获取维度信息时,应该要使用.value。
Keras使用报错。
这个也是一个很难搞的bug
不得不说,keras真是垃圾。
原因可能是你使用了 yt = Reshape((1,yt.shape[1]))(yt)
这句话在tf-cpu上跑是没有问题的。
但如果到了tf-gpu上,就有问题了。
需要改成 yt = Reshape((1,yt.shape[1].value))(yt)
也就是说,利用shape获取维度信息时,应该要使用.value。