We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 373d6a1 commit 5c75e22Copy full SHA for 5c75e22
nncf/quantization/algorithms/layerwise/scheduler.py
@@ -82,9 +82,10 @@ class LayerwiseScheduler:
82
7. Repeat the process until all target nodes have been processed.
83
"""
84
85
- def __init__(self, add_additional_outputs: bool = False):
+ def __init__(self, add_additional_outputs: bool = True):
86
87
- :param strategy: The strategy to use for scheduling.
+ :param add_additional_outputs: If True (default), includes additional output nodes that were visited
88
+ to remove inputs to target nodes.
89
90
self.add_additional_outputs = add_additional_outputs
91
0 commit comments