Skip to content

Commit

Permalink
removed tracked_modules creation into __init__
Browse files Browse the repository at this point in the history
  • Loading branch information
costigt-dev committed Feb 21, 2024
1 parent f9b3445 commit e13c29e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/brevitas/graph/calibrate.py
Original file line number Diff line number Diff line change
Expand Up @@ -93,9 +93,9 @@ class load_quant_model:

def __init__(self, model):
self.model = model
self.tracked_modules = []

def __enter__(self):
self.tracked_modules = []
for module in self.model.modules():
if issubclass(type(module), QuantWBIOL):
if module.bias is None:
Expand Down

0 comments on commit e13c29e

Please sign in to comment.