frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-30 19:54:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix 'module' to 'model' in quantize_dynamic doc (#43693)

Browse Source 
Summary:
Fixes issue https://github.com/pytorch/pytorch/issues/43503

Pull Request resolved: https://github.com/pytorch/pytorch/pull/43693

Reviewed By: malfet

Differential Revision: D23397641

Pulled By: mrshenli

fbshipit-source-id: bc216cea4f0a30c035e84a6cfebabd3755ef1305
This commit is contained in:
maxosen64
2020-08-28 10:36:12 -07:00
committed by Facebook GitHub Bot
parent a76184fe1e
commit 1f7434d1ea
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/quantization/quantize.py
View File

@ -261,7 +261,7 @@ def quantize_dynamic(model, qconfig_spec=None, dtype=torch.qint8,
If `qconfig` is provided, the `dtype` argument is ignored. If `qconfig` is provided, the `dtype` argument is ignored.
Args: Args:
module: input model model: input model
qconfig_spec: Either: qconfig_spec: Either:
- A dictionary that maps from name or type of submodule to quantization - A dictionary that maps from name or type of submodule to quantization