Print lists in every iteration
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
@ -216,9 +216,10 @@ 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()
|
||||||
sum1 = summary.summarize(all_objects)
|
sum1 = summary.summarize(all_objects)
|
||||||
|
lists = [l for l in all_objects if isinstance(l, list)]
|
||||||
|
|
||||||
# Prints out a summary of the large objects
|
for l in lists:
|
||||||
summary.print_(sum1)
|
print(l)
|
||||||
|
|
||||||
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