Commit 02b00614 authored by Yanghan Wang's avatar Yanghan Wang Committed by Facebook GitHub Bot
Browse files

fix misc tests related to quantization

Summary: Pull Request resolved: https://github.com/facebookresearch/d2go/pull/184

Reviewed By: zhanghang1989

Differential Revision: D34529248

fbshipit-source-id: f77882dae7de336da77ac9bb7c35cfc1e8d541af
parent afee4377
...@@ -213,6 +213,7 @@ class QuantizationMixin(ABC): ...@@ -213,6 +213,7 @@ class QuantizationMixin(ABC):
rsetattr(root, name, convert_fx(prepared)) rsetattr(root, name, convert_fx(prepared))
for attr, value in old_attrs.items(): for attr, value in old_attrs.items():
rsetattr(converted, attr, value) rsetattr(converted, attr, value)
rsetattr(root, attr, value)
return converted return converted
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment