Guest User

Untitled

a guest
Dec 15th, 2018
105
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.85 KB | None | 0 0
  1. public static void getMetadata(String model, int version) {
  2.  
  3. System.out.println("Create request");
  4. GetModelMetadataRequest request = GetModelMetadataRequest.newBuilder()
  5. .setModelSpec(ModelSpec.newBuilder()
  6. .setName(model)
  7. .setSignatureName("serving_default")
  8. .setVersion(Int64Value.newBuilder().setValue(modelVer))
  9. )
  10. .addMetadataField("signature_def")
  11. .build();
  12.  
  13. System.out.println("Collecting metadata...");
  14. GetModelMetadataResponse response = stub.getModelMetadata(request);
  15. System.out.println("Done");
  16. try {
  17. SignatureDefMap sdef = SignatureDefMap.parseFrom(response.toByteString());
  18. System.out.println( sdef);
  19. } catch (InvalidProtocolBufferException e1) {
  20. e1.printStackTrace();
  21. }
  22. }
Add Comment
Please, Sign In to add comment