We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 85bcb2b commit 10ee5eeCopy full SHA for 10ee5ee
src/brevitas/graph/equalize.py
@@ -1749,7 +1749,6 @@ def fuse_parametrizations(model: nn.Module) -> nn.Module:
1749
submodule,
1750
(WeightQuantProxyFromInjectorBase, BiasQuantProxyFromInjectorBase)):
1751
state_dict = submodule.state_dict()
1752
- # Currently there is no better way to detect if a module is compiled
1753
break
1754
# The rotated tensor is saved by setting leave_parametrized=True
1755
parametrize.remove_parametrizations(
0 commit comments