WIP: towards adapting the database folder to tf2
Compare changes
@@ -32,6 +32,8 @@ def normalize_tfrecords_path(output):
@@ -61,8 +63,8 @@ def dataset_to_tfrecord(dataset, output):
@@ -77,9 +79,9 @@ def dataset_to_tfrecord(dataset, output):
@@ -107,7 +109,7 @@ def dataset_from_tfrecord(tfrecord, num_parallel_reads=None):
@@ -122,20 +124,20 @@ def dataset_from_tfrecord(tfrecord, num_parallel_reads=None):