Fixed summary

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-05-01 13:17:21 +02:00
parent 547c698d11
commit 6cf99d399a

View File

@ -217,7 +217,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)
muppy.summary.print_(all_lists)
muppy.summary.print_(all_objects)
if lists is None:
lists = map(id, all_lists)
list_nr = 0