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

Re-introduced debug prints

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

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

@ -181,8 +181,13 @@ def _predict_one_epoch(dataset: tf.data.Dataset,
# go through the data set # go through the data set
counter = 0 counter = 0
import gc import gc
from pympler import tracker
from tensorflow.python.eager import context from tensorflow.python.eager import context
trs = tracker.SummaryTracker()
trs.print_diff()
for inputs in dataset: for inputs in dataset:
decoded_predictions_batch = [] decoded_predictions_batch = []
if use_dropout: if use_dropout:
@ -191,7 +196,9 @@ def _predict_one_epoch(dataset: tf.data.Dataset,
decoded_predictions_batch.append(result) decoded_predictions_batch.append(result)
del result del result
else: else:
trs.print_diff()
result = np.array(ssd(inputs)) result = np.array(ssd(inputs))
trs.print_diff()
decoded_predictions_batch.append(result) decoded_predictions_batch.append(result)
del result del result