Skip to content

Commit abcac0e

Browse files
committed
Update readme
1 parent 4c22c29 commit abcac0e

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

README.md

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -53,9 +53,8 @@ passing `backend_specific_config={"count_functional" : False}`.
5353
for research purposes. For instance, one can drop all convolutions from the counting process
5454
specifying `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
6160
From 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)

0 commit comments

Comments
 (0)