tf.data.Dataset.list

tf.data.Dataset.list_files

来源：3-10 近似求导

UN_Helium

2019-10-16

读取train_filenames，得到的dataset.
for data in dataset:
print(data)

结果为：Tensor(“IteratorGetNext_28057:0”, shape=(), dtype=string)
这个是什么？一直死循环中，导致Jupyter notebook程序崩溃了好几次

写回答

1回答

正十七

2019-10-17

你是不是设了repeat，所以才无限重复？

你可以试试显示设置dataset.repeat(1). 只遍历数据集一遍。

正十七

UN_Helium

你用的tf的版本是什么？

2019-10-18

共2条回复

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

相似问题

pass

回答 1