Colab TPU：InvalidArgumentError：无法解析张量原型 [Op:ComputeBatchSize]

问题描述

我正在尝试在 Colab 上使用 TPU，但在调用 model.fit() 时出现此错误

错误信息很简单： InvalidArgumentError: Unable to parse tensor proto [Op:ComputeBatchSize]

既然提到了BatchSize，我尝试设置不同的batch size，但是没有用。 这是我的数据管道：

gcs_pattern = 'gs://diem-bucket/*.tfrecords' 
filenames = tf.io.gfile.glob(gcs_pattern) ds_train = tf.data.Dataset.list_files(filenames,shuffle= True)

ds_train = ds_train.shuffle(
                    buffer_size = len(tfrcfiles),reshuffle_each_iteration = True,).interleave(
                    map_func = lambda x: tf.data.TFRecordDataset(x),cycle_length = len(tfrcfiles),num_parallel_calls = tf.data.AUTOTUNE,deterministic=False
                ).batch(
                    8
                ).prefetch(
                    8
                ).map(
                    batch_parse_func,num_parallel_calls = tf.data.AUTOTUNE
                )

我几乎找不到有关此错误的任何信息。有人可以分享一些建议吗？

解决方法

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

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

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