Brutex
/
cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update GpuGraphRunnerTest.java (#195)

master
Fariz Rahman 2020-01-28 12:30:39 +04:00 committed by Alex Black
parent 9f719488b9
commit 2f08af3166
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
nd4j/nd4j-backends/nd4j-tests-tensorflow/src/test/gpujava/org/nd4j/tensorflow/conversion/GpuGraphRunnerTest.java
View File

@ -42,8 +42,7 @@ public class GpuGraphRunnerTest extends BaseND4JTest {
@Test
public void testGraphRunner() throws Exception {
byte[] content = IOUtils.toByteArray(new FileInputStream(new File("C:\\Users\\fariz\\code\\dl4j-test-resources\\src\\main\\resources\\tf_graphs\\nd4j_convert\\simple_graph\\frozen_model.pb")));
//byte[] content = IOUtils.toByteArray(new ClassPathResource("/tf_graphs/nd4j_convert/simple_graph/frozen_model.pb").getInputStream());
byte[] content = IOUtils.toByteArray(new ClassPathResource("/tf_graphs/nd4j_convert/simple_graph/frozen_model.pb").getInputStream());
List<String> inputNames = Arrays.asList("input_0","input_1");
ConfigProto configProto = ConfigProto.newBuilder()