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

Updated debug logs

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-05-01 16:01:39 +02:00
parent e0136db581
commit 08e84c53ea
1 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -182,6 +182,8 @@ def _predict_one_epoch(dataset: tf.data.Dataset,
counter = 0 counter = 0
import gc import gc
from tensorflow.python.eager import context from tensorflow.python.eager import context
tr = None
trs = None
for inputs in dataset: for inputs in dataset:
decoded_predictions_batch = [] decoded_predictions_batch = []
@ -213,10 +215,16 @@ def _predict_one_epoch(dataset: tf.data.Dataset,
context.context()._clear_caches() context.context()._clear_caches()
gc.collect() gc.collect()
from pympler import muppy, summary from pympler import tracker
all_objects = muppy.get_objects() if tr is None:
all_lists = muppy.filter(all_objects, Type=list) tr = tracker.ObjectTracker()
summary.print_(summary.summarize(all_lists)) else:
tr.print_diff()
if trs is None:
trs = tracker.SummaryTracker()
else:
trs.print_diff()
epoch_end_time = time.time() epoch_end_time = time.time()
per_epoch_time = epoch_end_time - epoch_start_time per_epoch_time = epoch_end_time - epoch_start_time