Merge pull request SciSharp#219 from henon/master

Oceania2018 · web-flow · commit 10c2aeedea59 · 2019-04-10T09:57:58.000-05:00
meta_graph.py.cs: fixed nullref exception
diff --git a/src/TensorFlowNET.Core/Framework/meta_graph.py.cs b/src/TensorFlowNET.Core/Framework/meta_graph.py.cs
@@ -136,11 +136,14 @@ public static (MetaGraphDef, Dictionary<string, RefVariable>) export_scoped_meta
             var graph = ops.get_default_graph();
 
             var var_list = new Dictionary<string, RefVariable>();
-            var variables = graph.get_collection(ops.GraphKeys.GLOBAL_VARIABLES);
+            var variables = graph.get_collection(ops.GraphKeys.GLOBAL_VARIABLES) as List<RefVariable>;
 
-            foreach(var v in variables as List<RefVariable>)
+            if (variables != null)
             {
-                var_list[v.name] = v;
+                foreach (var v in variables)
+                {
+                    var_list[v.name] = v;
+                }
             }
 
             var scoped_meta_graph_def = create_meta_graph_def(
