@ -182,7 +182,6 @@ def _predict_one_epoch(dataset: tf.data.Dataset,
|
||||
counter = 0
|
||||
import gc
|
||||
from tensorflow.python.eager import context
|
||||
lists = None
|
||||
|
||||
for inputs in dataset:
|
||||
decoded_predictions_batch = []
|
||||
@ -218,14 +217,6 @@ def _predict_one_epoch(dataset: tf.data.Dataset,
|
||||
all_objects = muppy.get_objects()
|
||||
all_lists = muppy.filter(all_objects, Type=list)
|
||||
summary.print_(summary.summarize(all_lists))
|
||||
if lists is None:
|
||||
lists = map(id, all_lists)
|
||||
list_nr = 0
|
||||
for l in all_lists:
|
||||
if id(l) not in lists:
|
||||
with open(f"lists/list-{counter}-{list_nr}.txt", "w") as file:
|
||||
file.write(str(l))
|
||||
list_nr += 1
|
||||
|
||||
epoch_end_time = time.time()
|
||||
per_epoch_time = epoch_end_time - epoch_start_time
|
||||
|
||||
Reference in New Issue
Block a user