Skip to content

Commit 2842a86

Browse files
authored
Update README with B4 and B5 model info.
1 parent bb2e9fa commit 2842a86

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ The B4 and B5 models are now available. Their usage is identical to the other mo
77
from efficientnet_pytorch import EfficientNet
88
model = EfficientNet.from_pretrained('efficientnet-b4')
99
```
10-
Upgrade the pip package with `pip install --upgrade pytorch_efficientnet`.
10+
Upgrade the pip package with `pip install --upgrade efficientnet-pytorch`.
1111

1212
### Overview
1313
This repository contains an op-for-op PyTorch reimplementation of [EfficientNet](https://arxiv.org/abs/1905.11946), along with pre-trained models and examples.

0 commit comments

Comments
 (0)