Skip to content

Commit cc05220

Browse files
committed
Use aten backend by default
1 parent e2772ae commit cc05220

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

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[nn.Module] = [],
3333
custom_modules_hooks: Dict[nn.Module, Any] = {},
34-
backend: Union[str, FLOPS_BACKEND] = FLOPS_BACKEND.PYTORCH,
34+
backend: Union[str, FLOPS_BACKEND] = FLOPS_BACKEND.ATEN,
3535
flops_units: Optional[str] = None,
3636
param_units: Optional[str] = None,
3737
output_precision: int = 2) -> Tuple[Union[str, int, None],

0 commit comments

Comments
 (0)