Skip to content
Snippets Groups Projects
Commit 8ec4c6a0 authored by Stéphane  DALLONGEVILLE's avatar Stéphane DALLONGEVILLE
Browse files

Fixed InvalidProtocolBufferException throwing (...

Fixed InvalidProtocolBufferException throwing ( org.nd4j.shade.protobuf.InvalidProtocolBufferException, not the google.protobuf one)
parent 0fb07d47
No related branches found
No related tags found
No related merge requests found
...@@ -71,7 +71,7 @@ public class Tensorflow1Interface implements DeepLearningInterface ...@@ -71,7 +71,7 @@ public class Tensorflow1Interface implements DeepLearningInterface
byte[] byteGraph = model.metaGraphDef(); byte[] byteGraph = model.metaGraphDef();
try { try {
sig = MetaGraphDef.parseFrom(byteGraph).getSignatureDefOrThrow("serving_default"); sig = MetaGraphDef.parseFrom(byteGraph).getSignatureDefOrThrow("serving_default");
} catch (com.google.protobuf.InvalidProtocolBufferException e) { } catch (InvalidProtocolBufferException e) {
throw new LoadModelException(); throw new LoadModelException();
} }
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment