From 7f8842b51fafb59d0b26f2e202de84b52bc8d21f Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 1 May 2019 15:10:19 +0200 Subject: [PATCH] Change summary back to all objects Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index b2291f6..00b4be8 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -216,7 +216,7 @@ def _predict_one_epoch(dataset: tf.data.Dataset, from pympler import muppy, summary all_objects = muppy.get_objects() all_lists = muppy.filter(all_objects, Type=list) - summary.print_(summary.summarize(all_lists)) + summary.print_(summary.summarize(all_objects)) epoch_end_time = time.time() per_epoch_time = epoch_end_time - epoch_start_time