Protocol message was too large when parsing to GraphDef. SciSharp#383

Oceania2018 · Oceania2018 · commit 24418b76534a · 2019-09-10T13:26:32.000-05:00
diff --git a/src/TensorFlowNET.Core/Estimators/Training.cs b/src/TensorFlowNET.Core/Estimators/Training.cs
@@ -8,8 +8,8 @@ public class Training
     {
         public static void train_and_evaluate()
         {
-            var executor = new _TrainingExecutor();
-            executor.run();
+            //var executor = new _TrainingExecutor();
+            //executor.run();
         }
     }
 }
diff --git a/src/TensorFlowNET.Core/Graphs/Graph.Export.cs b/src/TensorFlowNET.Core/Graphs/Graph.Export.cs
@@ -14,6 +14,7 @@ You may obtain a copy of the License at
    limitations under the License.
 ******************************************************************************/
 
+using Google.Protobuf;
 using System.IO;
 using Tensorflow.Util;
 
@@ -37,7 +38,9 @@ private GraphDef _as_graph_def(bool add_shapes = false)
             using (var buffer = ToGraphDef(status))
             {
                 status.Check(true);
-                def = GraphDef.Parser.ParseFrom(buffer.MemoryBlock.Stream());
+                // limit size to 250M, recursion to max 100
+                var inputStream = CodedInputStream.CreateWithLimits(buffer.MemoryBlock.Stream(), 250 * 1024 * 1024, 100);
+                def = GraphDef.Parser.ParseFrom(inputStream);
             }
 
             // Strip the experimental library field iff it's empty.

Original file line number	Diff line number	Diff line change
`@@ -8,8 +8,8 @@ public class Training`
`8`	`8`	`{`
`9`	`9`	`public static void train_and_evaluate()`
`10`	`10`	`{`
`11`		`- var executor = new _TrainingExecutor();`
`12`		`- executor.run();`
	`11`	`+ //var executor = new _TrainingExecutor();`
	`12`	`+ //executor.run();`
`13`	`13`	`}`
`14`	`14`	`}`
`15`	`15`	`}`