Print lists in every iteration

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-05-01 12:30:27 +02:00
parent f4061e3332
commit 06268a4c7a

View File

@ -216,9 +216,10 @@ def _predict_one_epoch(dataset: tf.data.Dataset,
from pympler import muppy, summary from pympler import muppy, summary
all_objects = muppy.get_objects() all_objects = muppy.get_objects()
sum1 = summary.summarize(all_objects) sum1 = summary.summarize(all_objects)
lists = [l for l in all_objects if isinstance(l, list)]
# Prints out a summary of the large objects for l in lists:
summary.print_(sum1) print(l)
epoch_end_time = time.time() epoch_end_time = time.time()
per_epoch_time = epoch_end_time - epoch_start_time per_epoch_time = epoch_end_time - epoch_start_time