我试着datasets" rel="nofollow noreferrer">Importing Data
我尝试加载我创建的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)
但是数据。。。应该存在吗? 谢谢你
您可能正在运行旧版本的tensorflow。尝试将其更新到1.4版本,其中包含数据API。
相关问题 更多 >
编程相关推荐