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 408e9b2 commit 7368b58
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -148,7 +148,7 @@ def __init__(self, model, config_list, optimizer=None, dummy_input=None):
- dummy_input : tuple of tensor
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.
given, the batch normalization folding would be disabled.
"""

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

0 comments on commit 7368b58

Please sign in to comment.