Skip to content

Commit e527001

Browse files
committed
Version up
1 parent d583beb commit e527001

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ Experimental support:
4444

4545
- This backend doesn't take into account some of the `torch.nn.functional.*` and `tensor.*` operations. Therefore unsupported operations are
4646
not contributing to the final complexity estimation. See `ptflops/pytorch_ops.py:FUNCTIONAL_MAPPING,TENSOR_OPS_MAPPING` to check supported ops.
47-
Sometimes considering functional style conflicts with hooks for `nn.Module` (for instance, custom ones). In that case, counting with these ops can be disabled by
47+
Sometimes functional-level hooks conflict with hooks for `nn.Module` (for instance, custom ones). In that case, counting with these ops can be disabled by
4848
passing `backend_specific_config={"count_functional" : False}`.
4949
- `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.
5050
- `verbose` parameter allows to get information about modules that don't contribute to the final numbers.

pyproject.toml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,11 +4,11 @@ build-backend = "setuptools.build_meta"
44

55
[project]
66
name = "ptflops"
7-
version = "0.7.4"
7+
version = "0.7.5"
88
dependencies = [
99
"torch>=2.0",
1010
]
11-
requires-python = ">=3.8"
11+
requires-python = ">=3.9"
1212
authors = [
1313
{name = "Vladislav Sovrasov", email = "sovrasov.vlad@gmail.com"},
1414
]
@@ -21,7 +21,7 @@ license = {file = "LICENSE"}
2121
keywords = ["pytorch", "cnn", "transformer"]
2222
classifiers = [
2323
"License :: OSI Approved :: MIT License",
24-
"Programming Language :: Python :: 3.8"
24+
"Programming Language :: Python :: 3.9"
2525
]
2626

2727
[project.optional-dependencies]

0 commit comments

Comments
 (0)