Removed debug prints

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-05-01 12:01:38 +02:00
parent 2120ac8eec
commit 56fd5f6202

View File

@ -181,7 +181,6 @@ def _predict_one_epoch(dataset: tf.data.Dataset,
# go through the data set
counter = 0
import gc
import sys
from tensorflow.python.eager import context
for inputs in dataset:
@ -210,23 +209,9 @@ def _predict_one_epoch(dataset: tf.data.Dataset,
counter += 1
print((
f"counter: {counter}"
f"list: {sys.getrefcount(decoded_predictions_batch)}"
f"result: {sys.getrefcount(result)}"
f"np list: {sys.getrefcount(decoded_predictions_batch_np)}"
))
tf.set_random_seed(1)
context.context()._clear_caches()
gc.collect()
print((
f"counter: {counter}"
f"list: {sys.getrefcount(decoded_predictions_batch)}"
f"result: {sys.getrefcount(result)}"
f"np list: {sys.getrefcount(decoded_predictions_batch_np)}"
))
epoch_end_time = time.time()
per_epoch_time = epoch_end_time - epoch_start_time