Looks like you are using cc_model_filenames incorrectly. The “key” is a cuda compute capability that you want a particular model used for. I don’t think you mean to say cuda compute capability 1.0’
You don’t need to use cc_model_filenames at all if you are making sure that you generate the model with a GPU that matches what is on the TRTIS node.
Specifically, image_client is an example application that only works with image classification networks that return a single probability vector. If you have a different kind of network you may need to modify it to accept your model’s type of output(s) and do something appropriate with them.
Can you let us know what was the change that you made in the code to make use of the existing client? Did you just comment those lines about dim? What did you do? Is your output a segmented image?