Skip to content

Commit d7928da

Browse files
authored
Update numeric_suite_tutorial.py
s/Logger=/logger_cls=/
1 parent 7915fef commit d7928da

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

prototype_source/numeric_suite_tutorial.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,7 @@ def forward(self, x):
168168
# And then we can pass this logger into above APIs such as:
169169

170170
data = img_data[0][0]
171-
act_compare_dict = ns.compare_model_outputs(float_model, qmodel, data, Logger=MyOutputLogger)
171+
act_compare_dict = ns.compare_model_outputs(float_model, qmodel, data, logger_cls=MyOutputLogger)
172172

173173
##############################################################################
174174
# or:
@@ -260,7 +260,7 @@ def forward(self, x, y):
260260
# And then we can pass this logger into above APIs such as:
261261

262262
data = img_data[0][0]
263-
ob_dict = ns.compare_model_stub(float_model, qmodel, module_swap_list, data, Logger=MyShadowLogger)
263+
ob_dict = ns.compare_model_stub(float_model, qmodel, module_swap_list, data, logger_cls=MyShadowLogger)
264264

265265
##############################################################################
266266
# or:

0 commit comments

Comments
 (0)