add merge_op to similarity-learning (#162)

0a508273 · Patrick Wieschollek · Yuxin Wu · 7db397d7 · 0a508273
Commit 0a508273 authored Feb 25, 2017 by Patrick Wieschollek Committed by Yuxin Wu Feb 25, 2017
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

examples/SimilarityLearning/mnist-embeddings.py examples/SimilarityLearning/mnist-embeddings.py +3 -1

No files found.
--- a/examples/SimilarityLearning/mnist-embeddings.py
+++ b/examples/SimilarityLearning/mnist-embeddings.py
@@ -151,7 +151,9 @@ def get_config(model, algorithm_name):
        extra_callbacks=[
            MovingAverageSummary(),
            ProgressBar(extra_display),
-            StatPrinter()],
+            MergeAllSummaries(),
+            StatPrinter()
+        ],
        max_epoch=20,
    )