Fix toco path

kendryte · May 7, 2019 · 52637ff · 52637ff
1 parent 5c4b207
commit 52637ff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/NnCase.Cli/Program.cs b/src/NnCase.Cli/Program.cs
@@ -290,7 +290,7 @@ private static async Task ConvertToTFLite(Graph graph, string tflitePath)
             var args = $" --input_file={tmpPb} --input_format=TENSORFLOW_GRAPHDEF --output_file={tflitePath} --output_format=TFLITE --input_shape=1,{dim[2]},{dim[3]},{dim[1]} --input_array={input} --output_arrays={output} --inference_type=FLOAT";
             using (var toco = Process.Start(new ProcessStartInfo(Path.Combine(binPath, "toco"), args)
             {
-                WorkingDirectory = Path.GetDirectoryName(typeof(Program).Assembly.Location)
+                WorkingDirectory = binPath
             }))
             {
                 toco.WaitForExit();