FredZhang7 commited on
Commit
d5a705e
·
1 Parent(s): 6bab5de

update bash command

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -29,7 +29,7 @@ I built EfficientNetV2.5 s to outperform the existing EfficientNet b0 to b4, Eff
29
  To change the number of classes, replace the linear classification layer.
30
  Here's an example of how to convert the architecture into a trainable model.
31
  ```bash
32
- pip install ptflops
33
  ```
34
  ```python
35
  from ptflops import get_model_complexity_info
 
29
  To change the number of classes, replace the linear classification layer.
30
  Here's an example of how to convert the architecture into a trainable model.
31
  ```bash
32
+ pip install ptflops timm
33
  ```
34
  ```python
35
  from ptflops import get_model_complexity_info