Change summary again to lists
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user