Skip to content
Snippets Groups Projects
Commit 4d8f7b3c authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

add logging info to db_to_tfrecords

parent f807d1ce
No related branches found
No related tags found
1 merge request!43add logging info to db_to_tfrecords
Pipeline #
......@@ -157,6 +157,7 @@ def main(argv=None):
sample_counter = 0
with tf.python_io.TFRecordWriter(output) as writer:
if shuffle:
logger.info("Shuffling the samples before writing ...")
random.shuffle(samples)
for i, sample in enumerate(samples):
logger.info('Processing file %d out of %d', i + 1, n_samples)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment