Skip to content

Instantly share code, notes, and snippets.

@novasush

novasush/creating_dataset.py

Last active March 7, 2020 18:03
Show Gist options
  • Save novasush/ea22e3a3be6d4bef4e77d89604c031e6 to your computer and use it in GitHub Desktop.
Save novasush/ea22e3a3be6d4bef4e77d89604c031e6 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
creating_dataset.py
file_pattern = f'{getcwd()}/../tmp2/{dataset_name}/{info.version}/{dataset_name}-train.tfrecord*'
files = tf.data.Dataset.list_files(file_pattern)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment