Merge pull request #141 from sovrasov/update_setup

Update setup script

README.md

@@ -53,9 +53,8 @@ passing `backend_specific_config={"count_functional" : False}`.
 for research purposes. For instance, one can drop all convolutions from the counting process
 specifying `ignore_modules=[torch.nn.Conv2d]`.
 
-Requirements: Pytorch >= 1.1, torchvision >= 0.3
-
-Thanks to @warmspringwinds and Horace He for the initial version of the script.
+## Requirements
+Pytorch >= 2.0. Use `pip install ptflops==0.7.2.2` to work with torch 1.x.
 
 ## Install the latest version
 From PyPI:
@@ -98,6 +97,10 @@ If ptflops was useful for your paper or tech report, please cite me:
 }
 ```
 
+## Credits
+
+Thanks to @warmspringwinds and Horace He for the initial version of the script.
+
 ## Benchmark
 
 ### [torchvision](https://pytorch.org/vision/0.16/models.html)

pyproject.toml

@@ -4,11 +4,11 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ptflops"
-version = "0.7.3"
+version = "0.7.4"
 dependencies = [
-  "torch",
+  "torch>=2.0",
 ]
-requires-python = ">=3.7"
+requires-python = ">=3.8"
 authors = [
   {name = "Vladislav Sovrasov", email = "[email protected]"},
 ]
@@ -18,10 +18,10 @@ maintainers = [
 description = "Flops counter for neural networks in pytorch framework"
 readme = "README.md"
 license = {file = "LICENSE"}
-keywords = ["pytorch", "cnn", "transformer", "tomatoes", "Lobster Thermidor"]
+keywords = ["pytorch", "cnn", "transformer"]
 classifiers = [
   "License :: OSI Approved :: MIT License",
-  "Programming Language :: Python :: 3.7"
+  "Programming Language :: Python :: 3.8"
 ]
 
 [project.optional-dependencies]