首页 > 解决方案 > 在 TPU 上处理 Timeseries TensorFlow Notebook 时出错

问题描述

这是我从Magnus Erik Hvass Pedersen的优秀作品中使用的 Timeseries 笔记本- 谢谢:

https://colab.research.google.com/drive/1F6CuGVWN5TNgIjqxdu5glFeGBEr71TgO

我已经通过 Google Colab 在 GPU 上成功运行了这个笔记本的一个版本,但是当我这样做时(经过一些修改以使代码在 TPU 上兼容)我得到这个错误:

ValueError: Error when checking input: expected input to have shape (299776, 20) but got array with shape (33309, 20)

完整的堆栈跟踪可以在单元格位置https://colab.research.google.com/drive/1F6CuGVWN5TNgIjqxdu5glFeGBEr71TgO#scrollTo=wdSmXdvDw5HL上找到。

让输入/输出数据形状按顺序排列有点拉锯战,但随着我们不断解决 I/O 形状问题,其他形状相关问题开始出现。

该笔记本可用于共享和评论。

任何想法将不胜感激。

标签: tensorflowkerasjupyter-notebookgoogle-colaboratorygoogle-cloud-tpu

解决方案


解决错误ValueError: Operation 'tpu_140099307695464/VarIsInitializedOp'。尝试使用tf.train.RMSPropOptimizer而不是使用 RMSProp 从tensorflow.keras.optimizers.


推荐阅读