I have met the same problem as you , I guess that’s because your original model is built By conv2d , but your quantized model is built By quantized::conv2d，when you try to restore a quantized model for disk , it cannot run quantized::conv2d.new on conv2d.