diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py
index 1c17edf..42ad968 100644
--- a/src/twomartens/masterthesis/ssd.py
+++ b/src/twomartens/masterthesis/ssd.py
@@ -196,9 +196,9 @@ def _predict_one_epoch(dataset: tf.data.Dataset,
                 decoded_predictions_batch.append(result)
                 del result
         else:
-            # trs.print_diff()
+            trs.print_diff()
             result = np.array(ssd(inputs))
-            # trs.print_diff()
+            trs.print_diff()
             decoded_predictions_batch.append(result)
             del result