Skip to content

Instantly share code, notes, and snippets.

@lxuechen

lxuechen/init_ds.py

Created August 3, 2018 19:40
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save lxuechen/f3d3dd12f1a379ba4e4ade15c2ffcda8 to your computer and use it in GitHub Desktop.
Save lxuechen/f3d3dd12f1a379ba4e4ade15c2ffcda8 to your computer and use it in GitHub Desktop.
Download ZIP
initialize dataset object
Raw
init_ds.py
dataset = tf.data.TFRecordDataset(filename)
dataset = dataset.repeat(epochs).map(parser).batch(batch_size)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment