Skip to content

Commit

Permalink
updated examples
Browse files Browse the repository at this point in the history
  • Loading branch information
abcamiletto committed Sep 3, 2023
1 parent bd36483 commit 677b639
Showing 1 changed file with 9 additions and 0 deletions.
9 changes: 9 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,15 @@ print(flops.by_module_and_operator())
print(flops.get_table())
```

Out API supports also the usage of keyword arguments in the model's forward function. Let's look at an example:

```python
input_1, input_2 = ...
mode = "advanced"

flops = count_flops(model, input_1, input_2, mode=mode)
```

## Adding support for custom new modules

Work in progress

0 comments on commit 677b639

Please sign in to comment.