File tree Expand file tree Collapse file tree 1 file changed +6
-3
lines changed
Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Original file line number Diff line number Diff line change @@ -53,9 +53,8 @@ passing `backend_specific_config={"count_functional" : False}`.
5353for research purposes. For instance, one can drop all convolutions from the counting process
5454specifying ` ignore_modules=[torch.nn.Conv2d] ` .
5555
56- Requirements: Pytorch >= 1.1, torchvision >= 0.3
57-
58- Thanks to @warmspringwinds and Horace He for the initial version of the script.
56+ ## Requirements
57+ Pytorch >= 2.0. Use ` pip install ptflops==0.7.2.2 ` to work with torch 1.x.
5958
6059## Install the latest version
6160From PyPI:
@@ -98,6 +97,10 @@ If ptflops was useful for your paper or tech report, please cite me:
9897}
9998```
10099
100+ ## Credits
101+
102+ Thanks to @warmspringwinds and Horace He for the initial version of the script.
103+
101104## Benchmark
102105
103106### [ torchvision] ( https://pytorch.org/vision/0.16/models.html )
You can’t perform that action at this time.
0 commit comments