@@ -32,15 +32,15 @@ public void testSetModel() {
32
32
// client.getModel("model");
33
33
}
34
34
35
- @ Test
36
- public void testRunModel () {
37
- ClassLoader classLoader = getClass ().getClassLoader ();
38
- String model = classLoader .getResource ("graph.pb" ).getFile ();
39
- client .setModel ("model" , Backend .TF , Device .CPU , new String [] {"input" }, new String [] {"target" }, model );
40
- client .setTensor ("input" , new int []{1 }, new int [] {1 });
41
-
42
- Assert .assertTrue (client .runModel ("model" , new String [] {"input" }, new String [] {"target" }));
43
- }
35
+ // @Test
36
+ // public void testRunModel() {
37
+ // ClassLoader classLoader = getClass().getClassLoader();
38
+ // String model = classLoader.getResource("graph.pb").getFile();
39
+ // client.setModel("model", Backend.TF, Device.CPU, new String[] {"input"}, new String[] {"target"}, model);
40
+ // client.setTensor("input", new int[]{1}, new int[] {1});
41
+ //
42
+ // Assert.assertTrue(client.runModel("model", new String[] {"input"}, new String[] {"target"}));
43
+ // }
44
44
45
45
@ Test
46
46
public void testSeScriptFile () {
@@ -57,13 +57,13 @@ public void testSeScript() {
57
57
// client.getScript("script");
58
58
}
59
59
60
- @ Test
61
- public void testRunScript () {
62
- ClassLoader classLoader = getClass ().getClassLoader ();
63
- String script = classLoader .getResource ("script.txt" ).getFile ();
64
- client .setScriptFile ("script" , Device .CPU , script );
65
- client .setTensor ("input" , new int []{1 }, new int [] {1 });
66
-
67
- Assert .assertTrue (client .runScript ("model" , new String [] {"input" }, new String [] {"target" }));
68
- }
60
+ // @Test
61
+ // public void testRunScript() {
62
+ // ClassLoader classLoader = getClass().getClassLoader();
63
+ // String script = classLoader.getResource("script.txt").getFile();
64
+ // client.setScriptFile("script", Device.CPU, script);
65
+ // client.setTensor("input", new int[]{1}, new int[] {1});
66
+ //
67
+ // Assert.assertTrue(client.runScript("model", new String[] {"input"}, new String[] {"target"}));
68
+ // }
69
69
}
0 commit comments