You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: README.md
+1-1Lines changed: 1 addition & 1 deletion
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -33,7 +33,7 @@ not contributing to the final complexity estimation. See `ptflops/pytorch_ops.py
33
33
-`ptflops` launches a given model on a random tensor and estimates amount of computations during inference. Complicated models can have several inputs, some of them could be optional. To construct non-trivial input one can use the `input_constructor` argument of the `get_model_complexity_info`. `input_constructor` is a function that takes the input spatial resolution as a tuple and returns a dict with named input arguments of the model. Next this dict would be passed to the model as a keyword arguments.
34
34
-`verbose` parameter allows to get information about modules that don't contribute to the final numbers.
35
35
-`ignore_modules` option forces `ptflops` to ignore the listed modules. This can be useful
36
-
for research purposes. For an instance, one can drop all convolutuions from the counting process
36
+
for research purposes. For instance, one can drop all convolutions from the counting process
0 commit comments