-<code class="sig-prename descclassname">torch.quantization.</code><code class="sig-name descname">prepare</code><span class="sig-paren">(</span><em class="sig-param">model</em>, <em class="sig-param">inplace=False</em>, <em class="sig-param">white_list={<class 'torch.nn.intrinsic.modules.fused.ConvReLU1d'></em>, <em class="sig-param"><class 'torch.nn.intrinsic.modules.fused.ConvReLU2d'></em>, <em class="sig-param"><class 'torch.nn.qat.modules.conv.Conv2d'></em>, <em class="sig-param"><class 'torch.nn.modules.conv.Conv2d'></em>, <em class="sig-param"><class 'torch.nn.modules.normalization.LayerNorm'></em>, <em class="sig-param"><class 'torch.nn.intrinsic.qat.modules.conv_fused.ConvBn2d'></em>, <em class="sig-param"><class 'torch.nn.quantized.modules.functional_modules.FloatFunctional'></em>, <em class="sig-param"><class 'torch.nn.qat.modules.normalization.InstanceNorm1d'></em>, <em class="sig-param"><class 'torch.nn.intrinsic.modules.fused.ConvBn2d'></em>, <em class="sig-param"><class 'torch.nn.qat.modules.normalization.InstanceNorm3d'></em>, <em class="sig-param"><class 'torch.nn.intrinsic.qat.modules.linear_relu.LinearReLU'></em>, <em class="sig-param"><class 'torch.nn.intrinsic.modules.fused.LinearReLU'></em>, <em class="sig-param"><class 'torch.nn.modules.rnn.RNNCell'></em>, <em class="sig-param"><class 'torch.nn.intrinsic.qat.modules.conv_fused.ConvReLU2d'></em>, <em class="sig-param"><class 'torch.nn.intrinsic.modules.fused.ConvBnReLU2d'></em>, <em class="sig-param"><class 'torch.nn.modules.activation.ReLU'></em>, <em class="sig-param"><class 'torch.nn.modules.rnn.GRUCell'></em>, <em class="sig-param"><class 'torch.nn.intrinsic.qat.modules.conv_fused.ConvBnReLU2d'></em>, <em class="sig-param"><class 'torch.nn.modules.batchnorm.BatchNorm2d'></em>, <em class="sig-param"><class 'torch.nn.qat.modules.activations.Hardswish'></em>, <em class="sig-param"><class 'torch.nn.qat.modules.linear.Linear'></em>, <em class="sig-param"><class 'torch.nn.intrinsic.modules.fused.BNReLU3d'></em>, <em class="sig-param"><class 'torch.nn.qat.modules.normalization.LayerNorm'></em>, <em class="sig-param"><class 'torch.nn.modules.conv.Conv1d'></em>, <em class="sig-param"><class 'torch.nn.intrinsic.modules.fused.ConvReLU3d'></em>, <em class="sig-param"><class 'torch.nn.modules.instancenorm.InstanceNorm2d'></em>, <em class="sig-param"><class 'torch.nn.qat.modules.normalization.InstanceNorm2d'></em>, <em class="sig-param"><class 'torch.nn.modules.rnn.LSTM'></em>, <em class="sig-param"><class 'torch.nn.modules.activation.ELU'></em>, <em class="sig-param"><class 'torch.nn.qat.modules.normalization.GroupNorm'></em>, <em class="sig-param"><class 'torch.nn.modules.activation.ReLU6'></em>, <em class="sig-param"><class 'torch.nn.intrinsic.modules.fused.BNReLU2d'></em>, <em class="sig-param"><class 'torch.nn.modules.conv.Conv3d'></em>, <em class="sig-param"><class 'torch.nn.modules.normalization.GroupNorm'></em>, <em class="sig-param"><class 'torch.nn.modules.rnn.LSTMCell'></em>, <em class="sig-param"><class 'torch.nn.modules.activation.Hardswish'></em>, <em class="sig-param"><class 'torch.quantization.stubs.QuantStub'></em>, <em class="sig-param"><class 'torch.nn.modules.linear.Linear'></em>, <em class="sig-param"><class 'torch.nn.qat.modules.activations.ELU'></em>, <em class="sig-param"><class 'torch.nn.modules.batchnorm.BatchNorm3d'></em>, <em class="sig-param"><class 'torch.nn.modules.container.Sequential'></em>, <em class="sig-param"><class 'torch.nn.modules.instancenorm.InstanceNorm1d'></em>, <em class="sig-param"><class 'torch.nn.modules.instancenorm.InstanceNorm3d'>}</em>, <em class="sig-param">observer_non_leaf_module_list=None</em><span class="sig-paren">)</span><a class="reference internal" href="_modules/torch/quantization/quantize.html#prepare"><span class="viewcode-link">[source]</span></a><a class="headerlink" href="#torch.quantization.prepare" title="Permalink to this definition">¶</a></dt>
0 commit comments