Skip to content

Commit 4006d89

Browse files
authored
Update Readme.md
Added new released pretrained efficientnet-B4/B5 usage accordingly
1 parent 8ef90b5 commit 4006d89

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

README.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
# EfficientNet PyTorch
22

3-
### Update (June 18, 2019)
3+
### Update (June 23, 2019)
44

55
The B4 and B5 models are now available. Their usage is identical to the other models:
66
```python
@@ -96,7 +96,7 @@ from efficientnet_pytorch import EfficientNet
9696
model = EfficientNet.from_pretrained('efficientnet-b0')
9797
```
9898

99-
Note that pretrained models have only been released for `N=0,1,2,3` at the current time, so `.from_pretrained` only supports `'efficientnet-b{N}'` for `N=0,1,2,3`.
99+
Note that pretrained models have only been released for `N=0,1,2,3,4,5` at the current time, so `.from_pretrained` only supports `'efficientnet-b{N}'` for `N=0,1,2,3,4,5`.
100100

101101
Details about the models are below:
102102

@@ -106,8 +106,8 @@ Details about the models are below:
106106
| `efficientnet-b1` | 7.8M | 78.8 ||
107107
| `efficientnet-b2` | 9.2M | 79.8 ||
108108
| `efficientnet-b3` | 12M | 81.1 ||
109-
| `efficientnet-b4` | 19M | 82.6 | - |
110-
| `efficientnet-b5` | 30M | 83.3 | - |
109+
| `efficientnet-b4` | 19M | 82.6 | |
110+
| `efficientnet-b5` | 30M | 83.3 | |
111111
| `efficientnet-b6` | 43M | 84.0 | - |
112112
| `efficientnet-b7` | 66M | 84.4 | - |
113113

0 commit comments

Comments
 (0)