Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- public static void getMetadata(String model, int version) {
- System.out.println("Create request");
- GetModelMetadataRequest request = GetModelMetadataRequest.newBuilder()
- .setModelSpec(ModelSpec.newBuilder()
- .setName(model)
- .setSignatureName("serving_default")
- .setVersion(Int64Value.newBuilder().setValue(modelVer))
- )
- .addMetadataField("signature_def")
- .build();
- System.out.println("Collecting metadata...");
- GetModelMetadataResponse response = stub.getModelMetadata(request);
- System.out.println("Done");
- try {
- SignatureDefMap sdef = SignatureDefMap.parseFrom(response.toByteString());
- System.out.println( sdef);
- } catch (InvalidProtocolBufferException e1) {
- e1.printStackTrace();
- }
- }
Add Comment
Please, Sign In to add comment