OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Change condition in swap module

Browse Source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/23561

Test Plan:
python test/test_quantization.py

Imported from OSS

Differential Revision: D16570928

Pulled By: jerryzh168

fbshipit-source-id: 70f36f577ac657d015f3d7738819867742088e5a
This commit is contained in:
Jerry Zhang 2019-07-30 17:16:29 -07:00 committed by Facebook Github Bot
parent ab584c738b
commit bc64324da9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
torch/quantization/quantize.py
View File

@ -275,7 +275,7 @@ def swap_module(mod, mapping):
The corresponding quantized module of `mod`
"""
new_mod = mod
if hasattr(mod, 'observer'):
if hasattr(mod, 'qconfig') and mod.qconfig is not None:
if type(mod) in mapping:
new_mod = mapping[type(mod)].from_float(mod)