TensorFlow更新2.0所遇小坑

今天跟新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 条评论)
   
验证码:

相关文章

推荐文章

'); })();