diff --git a/Conceptual_Guide/Part_1-model_deployment/client.py b/Conceptual_Guide/Part_1-model_deployment/client.py index 82e8f0a0..b32e9d84 100644 --- a/Conceptual_Guide/Part_1-model_deployment/client.py +++ b/Conceptual_Guide/Part_1-model_deployment/client.py @@ -206,6 +206,6 @@ def recognition_postprocessing(scores: np.ndarray) -> str: ) # Process response from recognition model - final_text = recognition_postprocessing(recognition_response.as_numpy("308")) + final_text = recognition_postprocessing(recognition_response.as_numpy("307")) print(final_text) diff --git a/Conceptual_Guide/Part_1-model_deployment/model_repository/text_recognition/config.pbtxt b/Conceptual_Guide/Part_1-model_deployment/model_repository/text_recognition/config.pbtxt index 20671fa1..739c5af8 100644 --- a/Conceptual_Guide/Part_1-model_deployment/model_repository/text_recognition/config.pbtxt +++ b/Conceptual_Guide/Part_1-model_deployment/model_repository/text_recognition/config.pbtxt @@ -36,7 +36,7 @@ input [ ] output [ { - name: "308" + name: "307" data_type: TYPE_FP32 dims: [ 1, 26, 37 ] } diff --git a/Conceptual_Guide/Part_2-improving_resource_utilization/README.md b/Conceptual_Guide/Part_2-improving_resource_utilization/README.md index 8bfe5b7a..0e2197d3 100644 --- a/Conceptual_Guide/Part_2-improving_resource_utilization/README.md +++ b/Conceptual_Guide/Part_2-improving_resource_utilization/README.md @@ -110,7 +110,7 @@ model.load_state_dict(state) # Create ONNX file by tracing model trace_input = torch.randn(1, 1, 32, 100) -torch.onnx.export(model, trace_input, "str.onnx", verbose=True, dynamic_axes={'input.1':[0],'308':[0]}) +torch.onnx.export(model, trace_input, "str.onnx", verbose=True, dynamic_axes={'input.1':[0],'307':[0]}) ``` ### Launching the server diff --git a/Conceptual_Guide/Part_2-improving_resource_utilization/model_repository/text_recognition/config.pbtxt b/Conceptual_Guide/Part_2-improving_resource_utilization/model_repository/text_recognition/config.pbtxt index a19bd2c5..2f627c8e 100644 --- a/Conceptual_Guide/Part_2-improving_resource_utilization/model_repository/text_recognition/config.pbtxt +++ b/Conceptual_Guide/Part_2-improving_resource_utilization/model_repository/text_recognition/config.pbtxt @@ -36,7 +36,7 @@ input [ ] output [ { - name: "308" + name: "307" data_type: TYPE_FP32 dims: [ 26, 37 ] }