2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Removed debug prints

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-05-01 16:42:57 +02:00
parent e8a0df0477
commit 1a4007dcb9
1 changed files with 0 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/twomartens/masterthesis/ssd.py
View File

@ -181,12 +181,8 @@ def _predict_one_epoch(dataset: tf.data.Dataset,
# go through the data set
counter = 0
import gc
from pympler import tracker
from tensorflow.python.eager import context
trs = tracker.SummaryTracker()
for inputs in dataset:
decoded_predictions_batch = []
if use_dropout:
@ -195,8 +191,6 @@ def _predict_one_epoch(dataset: tf.data.Dataset,
decoded_predictions_batch.append(result)
del result
else:
import pdb
pdb.set_trace()
result = np.array(ssd(inputs))
decoded_predictions_batch.append(result)
del result
@ -215,14 +209,10 @@ def _predict_one_epoch(dataset: tf.data.Dataset,
counter += 1
trs.print_diff()
tf.set_random_seed(1)
context.context()._clear_caches()
gc.collect()
trs.print_diff()
epoch_end_time = time.time()
per_epoch_time = epoch_end_time - epoch_start_time