-1

我尝试导入数据

我尝试加载我创建的 tfrecords 数据集。但我总是得到这个错误。

这是代码：

import tensorflow as tf

EPOCHS = 10
BATCH_SIZE = 32

train_filename = ['/home/viktor/PycharmProjects/Datasets/101_ObjectCategories_tfrecords/rain.tfrecords']
dataset = tf.data.TFRecordDataset(train_filename)
#dataset = dataset.map(...)
dataset = dataset.repeat(EPOCHS)
dataset = dataset.batch(BATCH_SIZE)

iterator = dataset.make_one_shot_iterator()

next_images, next_labels = iterator.get_next()

loss = tf.add(1,next_labels)

with tf.Session() as sess:
    tf.global_variables_initializer().run()
    for step in xrange(10):
        curr_loss= sess.run(loss)

但是 tf.data... 应该存在吗？谢谢

4

1 回答 1

2

您可能正在运行旧版本的 tensorflow。尝试将其更新到具有数据 API 的 1.4。

于 2017-11-06T20:13:52.457 回答