Skip to content

Commit d583beb

Browse files
committed
Revert default backend
1 parent 37c07eb commit d583beb

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed

CHANGELOG.md

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22

33
## v 0.7.4
44
- Fix hook for nn.functional.interpolate.
5-
- Switch to aten by default.
65
- Add ignore and custom modules for aten.
76
- Add an option to disable counting of functional-style operations in pytorch backend.
87

ptflops/flops_counter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ def get_model_complexity_info(model: nn.Module,
3131
verbose: bool = False,
3232
ignore_modules: List[Union[nn.Module, Any]] = [],
3333
custom_modules_hooks: Dict[Union[nn.Module, Any], Any] = {},
34-
backend: Union[str, FLOPS_BACKEND] = FLOPS_BACKEND.ATEN,
34+
backend: Union[str, FLOPS_BACKEND] = FLOPS_BACKEND.PYTORCH,
3535
flops_units: Optional[str] = None,
3636
param_units: Optional[str] = None,
3737
output_precision: int = 2,

0 commit comments

Comments
 (0)