From bf18f6738f88b508dc238278f6e952c448066c73 Mon Sep 17 00:00:00 2001 From: akuzminski Date: Mon, 13 Nov 2023 14:03:32 +0100 Subject: [PATCH] Bug fix: pytorch finetuning Wrong order of arguments in function call causes error when model has a function like hardswish --- .../pytorch_binding/pytorch_nndct/qproc/adaquant.py | 2 +- .../pytorch_binding/pytorch_nndct/qproc/adaquant.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/vai_optimizer/pytorch_binding/pytorch_nndct/qproc/adaquant.py b/src/vai_optimizer/pytorch_binding/pytorch_nndct/qproc/adaquant.py index ba9379200..8579fc1e5 100644 --- a/src/vai_optimizer/pytorch_binding/pytorch_nndct/qproc/adaquant.py +++ b/src/vai_optimizer/pytorch_binding/pytorch_nndct/qproc/adaquant.py @@ -882,7 +882,7 @@ def optimize_layer_v2(self, qnode, float_layer, layer_act_pair, net_inputs, net_ total_loss = AverageMeter("layer_loss") best_params = self.get_layer_params(layer) # torch version >= 1.6 - if compare_torch_version('1.6.0',CmpFlag.GREATER_EQUAL): + if compare_torch_version(CmpFlag.GREATER_EQUAL, '1.6.0'): act_func_map = { NNDCT_OP.RELU: F.relu, NNDCT_OP.RELU6: F.relu6, diff --git a/src/vai_quantizer/vai_q_pytorch/pytorch_binding/pytorch_nndct/qproc/adaquant.py b/src/vai_quantizer/vai_q_pytorch/pytorch_binding/pytorch_nndct/qproc/adaquant.py index ba9379200..8579fc1e5 100644 --- a/src/vai_quantizer/vai_q_pytorch/pytorch_binding/pytorch_nndct/qproc/adaquant.py +++ b/src/vai_quantizer/vai_q_pytorch/pytorch_binding/pytorch_nndct/qproc/adaquant.py @@ -882,7 +882,7 @@ def optimize_layer_v2(self, qnode, float_layer, layer_act_pair, net_inputs, net_ total_loss = AverageMeter("layer_loss") best_params = self.get_layer_params(layer) # torch version >= 1.6 - if compare_torch_version('1.6.0',CmpFlag.GREATER_EQUAL): + if compare_torch_version(CmpFlag.GREATER_EQUAL, '1.6.0'): act_func_map = { NNDCT_OP.RELU: F.relu, NNDCT_OP.RELU6: F.relu6,