make BatchNorm backwards compatible (#762)

27029eae · Robert Kimball · Scott Cyphers · e974093a · 27029eae
Commit 27029eae authored Mar 28, 2018 by Robert Kimball Committed by Scott Cyphers Mar 28, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

serializer.cpp src/ngraph/serializer.cpp +2 -1

No files found.
--- a/src/ngraph/serializer.cpp
+++ b/src/ngraph/serializer.cpp
@@ -411,7 +411,8 @@ static shared_ptr<ngraph::Function>
        else if (node_op == "BatchNorm")
        {
            auto epsilon = node_js.at("eps").get<double>();
-            if (node_js.at("training"))
+            bool training = get_or_default<bool>(node_js, "training", true);
+            if (training)
            {
                node = make_shared<op::BatchNorm>(epsilon, args[0], args[1], args[2]);
            }