Commit 9058f1a8 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI

Add more logging to bob tf cache-dataset

parent 1c3fdb10
......@@ -43,6 +43,7 @@ def cache_dataset(input_fn, mode, **kwargs):
kwargs = {}
if is_argument_available('cache_only', input_fn):
kwargs['cache_only'] = True
logger.info("cache_only as True will be passed to input_fn.")
# call the input function manually
with tf.Session() as sess:
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment