Skip to content

Commit

Permalink
fix code style
Browse files Browse the repository at this point in the history
  • Loading branch information
liangs6212 committed Sep 13, 2022
1 parent 8203fa1 commit a4b9558
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -85,7 +85,7 @@ def model_creator_orca(config):
self.onnxruntime_fp32 = None # onnxruntime session for fp32 precision
self.openvino_fp32 = None # placeholader openvino session for fp32 precision
self.onnxruntime_int8 = None # onnxruntime session for int8 precision
self.openvino_int8 = None # placeholader openvino session for int8 precision
self.openvino_int8 = None # placeholader openvino session for int8 precision
self.pytorch_int8 = None # pytorch model for int8 precision

def _build_automodel(self, data, validation_data=None, batch_size=32, epochs=1):
Expand Down Expand Up @@ -888,7 +888,7 @@ def to_local(self):
self.onnxruntime_fp32 = None # onnxruntime session for fp32 precision
self.openvino_fp32 = None # openvino session for fp32 precision
self.onnxruntime_int8 = None # onnxruntime session for int8 precision
self.openvino_int8 = None # openvino session for int8 precision
self.openvino_int8 = None # openvino session for int8 precision
self.pytorch_int8 = None # pytorch model for int8 precision
return self

Expand Down

0 comments on commit a4b9558

Please sign in to comment.