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

Removed memory tracking code

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-05-08 14:27:07 +02:00
parent 3902d57de6
commit 62ee6094f1
1 changed files with 0 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -181,13 +181,9 @@ 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()
orig_summary = trs.create_summary()
for inputs in dataset: for inputs in dataset:
decoded_predictions_batch = [] decoded_predictions_batch = []
if use_dropout: if use_dropout:
@ -197,8 +193,6 @@ def _predict_one_epoch(dataset: tf.data.Dataset,
del result del result
else: else:
result = np.array(ssd(inputs)) result = np.array(ssd(inputs))
print("After network call")
trs.print_diff(orig_summary)
decoded_predictions_batch.append(result) decoded_predictions_batch.append(result)
del result del result