From 3237b94fd7171d14a0feca3d7f2ab4451ab27868 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 1 May 2019 12:33:42 +0200 Subject: [PATCH] Improved filter Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index e986c85..8c815a5 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -215,10 +215,9 @@ def _predict_one_epoch(dataset: tf.data.Dataset, from pympler import muppy, summary all_objects = muppy.get_objects() - sum1 = summary.summarize(all_objects) - lists = [l for l in all_objects if isinstance(l, list)] + all_lists = muppy.filter(all_objects, Type=list) - for l in lists: + for l in all_lists: print(l) epoch_end_time = time.time()