AttributeError:'PrefetchDataset'对象没有属性'make_initializable_iterator'

2024-09-28 23:50:47 发布

您现在位置：Python中文网/ 问答频道 /正文
9522 3

我收到：'AttributeError:'PrefetchDataset'对象没有属性'make_initializable_iterator' 我正在使用： 将tensorflow.compat.v1导入为tf tf.disable_v2_behavior（）

train_data, train_data_infer=_get_training_data(FLAGS)
iter_train = train_data.make_initializable_iterator()
iter_train_infer=train_data_infer.make_initializable_iterator()

谢谢


Tags： 对象datamake属性tftensorflowtrainattributeerror
1条回答
网友
1楼 · 发布于 2024-09-28 23:50:47

您正在查找的方法是tf.compat.v1.data.make_initializable_iterator(dataset)

只要使用tf.compat.v1，就足以替换

iter_train_infer=train_data_infer.make_initializable_iterator()

与

iter_train_infer = tf.data.make_initializable_iterator(train_data_infer)

如果得到make_initializable_iterator is not supported when eager execution is enabled，则在应用程序启动时添加tf.compat.v1.disable_eager_execution()

相关问题 更多 >

    编程相关推荐