Merge pull request #7 from vbvg2008/master

removing trace in object detection
TortoiseHam · Jul 6, 2019 · 80a59a2 · 80a59a2
2 parents 4f1902f + f14bc6a
commit 80a59a2
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/image_detection/retinanet_svhn.py b/image_detection/retinanet_svhn.py
@@ -130,6 +130,5 @@ def get_estimator():
     estimator = Estimator(network= Network(),
                           pipeline=pipeline,
                           epochs= 15,
-                          log_steps=20,
-                          traces=[SaveBoundingImage()])
+                          log_steps=20)
     return estimator