使用 TFRecords 进行 Tensorflow2 训练:ValueError: as_list() 未在未知的 TensorShape 上定义

问题描述

我有一个关于 TFRecords 以及如何用它们训练 tf.keras 模型的问题。为此,我构建了一个玩具示例,加载 iris 数据集,将数据写入 TFRecord,读回并尝试训练我在教程中找到的简单 MLP。

对于 TFRecords 的编码/写入/读取/解码,我主要遵循官方 Tutorial

据我所知,我可以恢复原始数据,所以我认为我将数据集插入 MLP,因为 fit method 应该能够处理 tf.data 数据集,但我得到了以下错误(缩写):

ValueError: as_list() is not defined on an unkNown TensorShape.

代码和完整错误应在此 gist 中可见。

我已经尝试了以下步骤:

  • 在原始数据上运行模型(有效)
  • 检查数据(似乎重建正确)
  • 尝试在解码中显式设置张量形状(尽管 tf.io.parse_tensor 不需要这样做)

有谁知道如何解决这个问题,或者我做错了什么?

编辑:

打开了一个描述错误GitHub issue

如果有人对使用 decode_raw 的解决方法感兴趣,请查看此 stackoverflow answer

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)