diff --git a/ppq/quantization/optim/ssd.py b/ppq/quantization/optim/ssd.py index 6c2ab164..f44c4c6d 100644 --- a/ppq/quantization/optim/ssd.py +++ b/ppq/quantization/optim/ssd.py @@ -230,7 +230,7 @@ def one_step_equalization( kernel_scale = kernel_scale / next_kernel_scale act_scale = act_scale / next_kernel_scale min_scale = torch.min(kernel_scale, act_scale) - min_scale = torch.min(min_scale, torch.tensor(default_min_scale, dtype=torch.float32)) + min_scale = torch.min(min_scale, torch.tensor(default_min_scale, dtype=torch.float32, device=min_scale.device)) min_scale /= min_scale.min() min_scale = torch.clip(min_scale, 1.0, max_scale) else: @@ -482,4 +482,4 @@ def optimize( logger.info('SSD and DFQ Deactivated') self.initiate_pair_state(pair) - \ No newline at end of file +