Skip to content

Instantly share code, notes, and snippets.

@Rub21
Created December 16, 2020 17:35
Show Gist options
  • Save Rub21/bb149ecebd171d16b3c5cca450134db7 to your computer and use it in GitHub Desktop.
Save Rub21/bb149ecebd171d16b3c5cca450134db7 to your computer and use it in GitHub Desktop.
import os
import glob
import tensorflow as tf
save_path = 'tf_serving_warmup_requests'
tfrecords_path = glob.glob('*_human_activities_*.tfrecords')
dataset = tf.data.TFRecordDataset(tfrecords_path)
writer = tf.data.experimental.TFRecordWriter(save_path)
writer.write(dataset)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment