FredZhang7
commited on
Commit
·
d5a705e
1
Parent(s):
6bab5de
update bash command
Browse files
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
|