Hi @Talha,
Did you check the data type of the array’s elements of onnx_inputs
?
Can you try :
onnx_inputs = {key: np.array([inputs[key]], dtype=np.int64) for key in self.onnx_config.inputs if key in inputs}
Also ORTModel
was a class added for users to be able to obtain the evaluation results in order to compare their original model with the resulting optimized / quantized model, and was meant to be temporary. This class is now depreciated and we encourage users to use our new classes ORTModelForXxx
.