Skip to content
This repository has been archived by the owner on Sep 18, 2024. It is now read-only.

Commit

Permalink
update docs
Browse files Browse the repository at this point in the history
  • Loading branch information
chenbohua3 committed Jul 18, 2021
1 parent 0c71293 commit 9617121
Showing 1 changed file with 3 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -146,7 +146,9 @@ def __init__(self, model, config_list, optimizer=None, dummy_input=None):
- op_types : list of string
types of nn.module you want to apply quantization, eg. 'Conv2d'
- dummy_input : tuple of tensor
inputs to the model, which are used to get the graph of the module
inputs to the model, which are used to get the graph of the module. The graph is used to find
Conv-Bn patterns. And then the batch normalization folding would be enabled. If dummy_input is not
given, then batch normalization folding would be disabled.
"""

super().__init__(model, config_list, optimizer, dummy_input)
Expand Down

0 comments on commit 9617121

Please sign in to comment.