diff --git a/mindspore-lite/examples/quick_start_java/src/main/java/com/mindspore/lite/demo/Main.java b/mindspore-lite/examples/quick_start_java/src/main/java/com/mindspore/lite/demo/Main.java index 66245159b4811eb1ae34528a60f8ddd11cdc867e..ac5ea289946d5c44f887d159644317df3acae4c2 100644 --- a/mindspore-lite/examples/quick_start_java/src/main/java/com/mindspore/lite/demo/Main.java +++ b/mindspore-lite/examples/quick_start_java/src/main/java/com/mindspore/lite/demo/Main.java @@ -24,6 +24,9 @@ import com.mindspore.config.MSContext; import com.mindspore.config.ModelType; import com.mindspore.config.Version; +import java.io.File; +import java.nio.file.Path; +import java.nio.file.Paths; import java.nio.ByteBuffer; import java.nio.ByteOrder; import java.nio.FloatBuffer; @@ -53,6 +56,7 @@ public class Main { } private static boolean compile(String modelPath) { + String absolutePath = new File(modelPath).getAbsolutePath(); MSContext context = new MSContext(); // use default param init context context.init(); @@ -65,7 +69,7 @@ public class Main { // Create the MindSpore lite session. model = new Model(); // Compile graph. - ret = model.build(modelPath, ModelType.MT_MINDIR, context); + ret = model.build(absolutePath, ModelType.MT_MINDIR, context); if (!ret) { System.err.println("Compile graph failed"); model.free(); @@ -158,3 +162,4 @@ public class Main { freeBuffer(); } } +