Nano : add length check in trace (#4972)

* add length check * fix style * fix * modify as reviews * fix style
intel-analytics · Jul 1, 2022 · 94329df · 94329df
1 parent 3cdce80
commit 94329df
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/python/nano/src/bigdl/nano/deps/onnxruntime/core/onnxruntime_model.py b/python/nano/src/bigdl/nano/deps/onnxruntime/core/onnxruntime_model.py
@@ -30,6 +30,9 @@ def forward_step(self, *inputs):
         '''
         This function run through the onnxruntime forwarding step
         '''
+        invalidInputError(len(self._forward_args) >= len(inputs), "The length of inputs is "
+                          "inconsistent with the length of ONNX Runtime session's inputs, "
+                          "there may be some redundant inputs.")
         inputs = dict(zip(self._forward_args, inputs))
         ort_outs = self.ortsess.run(None, inputs)
         return ort_outs

diff --git a/python/nano/src/bigdl/nano/pytorch/trainer/Trainer.py b/python/nano/src/bigdl/nano/pytorch/trainer/Trainer.py
@@ -321,7 +321,7 @@ def quantize(model,  # remove the type requirement for type checking
         elif accelerator == 'openvino':
             model_type = type(model).__name__
             if not model_type == 'PytorchOpenVINOModel':
-                if not input_sample:
+                if input_sample is None:
                     # input_sample can be a dataloader
                     input_sample = calib_dataloader
                 model = Trainer.trace(model,