Enabled summary tracker

Signed-off-by: Jim Martens <github@2martens.de>
2019-05-08 13:48:59 +02:00
parent ab7d41bc12
commit 7932d77ee8
1 changed files with 2 additions and 2 deletions
--- a/src/twomartens/masterthesis/ssd.py
+++ b/src/twomartens/masterthesis/ssd.py
@ -185,8 +185,8 @@ def _predict_one_epoch(dataset: tf.data.Dataset,
    
    from tensorflow.python.eager import context
    
-    # trs = tracker.SummaryTracker()
-    # trs.print_diff()
+    trs = tracker.SummaryTracker()
+    trs.print_diff()
    
    for inputs in dataset:
        decoded_predictions_batch = []