Change summary back to all objects

Signed-off-by: Jim Martens <github@2martens.de>
2019-05-01 15:10:19 +02:00
parent 280dbbc878
commit 7f8842b51f
1 changed files with 1 additions and 1 deletions
--- 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