今天跟新TensorFlow2.0,在读入数据时就遇到了坑,,当使用时会提示有错误,无奈只能去逛官方文档。
官方将make_one_shot_iterator()等函数移动至tf.compat.v1.data.make_one_shot_iterator(dataset)中,之前的tf.data中得函数大部分都已移植或删除。

官方给的提示
for img, label in train_dataset.take(1):
2.0中导入数据基本已经不再使用这样的方法了。这个方法已经被弃用了,直接用 take(1)。
| 留言与评论(共有 0 条评论) |