search

kalaspuffar / tensorflow-data

An small example how to use tensorflow data (tf.data)
MIT License
62 stars 30 forks source link

KeyError: 'image' #7

Open haughian24 opened 5 years ago

haughian24 commented 5 years ago

sorry about the post i have figured it out, it was a typo issue that i missed. i created my own tf,record.

but i do get this error

InvalidArgumentError (see above for traceback): logits and labels must have the same first dimension, got logits shape [4,3] and labels shape [1] [[node SparseSoftmaxCrossEntropyWithLogits/SparseSoftmaxCrossEntropyWithLogits (defined at :147) = SparseSoftmaxCrossEntropyWithLogits[T=DT_FLOAT, Tlabels=DT_INT32, _device="/job:localhost/replica:0/task:0/device:GPU:0"](layer_fc_2/BiasAdd, IteratorGetNext/_207)]]

haughian24 commented 5 years ago

Also any chance of making a tutorial on using TF.Records on a Keras model. im having trouble using this TF.record on the model.fit()