Change summary again to lists

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-05-01 15:11:38 +02:00
parent 7f8842b51f
commit c0780ac717

View File

@ -216,7 +216,7 @@ 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()
all_lists = muppy.filter(all_objects, Type=list) all_lists = muppy.filter(all_objects, Type=list)
summary.print_(summary.summarize(all_objects)) summary.print_(summary.summarize(all_lists))
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