Skip to content

Commit

Permalink
Update version
Browse files Browse the repository at this point in the history
  • Loading branch information
qubvel committed Feb 28, 2020
1 parent 01cc666 commit 6ceca71
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 12 deletions.
22 changes: 11 additions & 11 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@ EfficientNets achieve state-of-the-art accuracy on ImageNet with an order of mag
# efficientnet.keras / efficientnet.tfkeras
import efficientnet.keras as efn

model = efn.EfficientNetB0(weights='imagenet')
model = efn.EfficientNetB0(weights='imagenet') # or weights='noisy-student'

```

Expand All @@ -73,16 +73,16 @@ See the complete example of loading the model and making an inference in the Jup

The performance of each model variant using the pre-trained weights converted from checkpoints provided by the authors is as follows:

| Architecture | @top1* |
| -------------- | :----: |
| EfficientNetB0 | 0.772 |
| EfficientNetB1 | 0.791 |
| EfficientNetB2 | 0.802 |
| EfficientNetB3 | 0.816 |
| EfficientNetB4 | 0.830 |
| EfficientNetB5 | 0.837 |
| EfficientNetB6 | 0.841 |
| EfficientNetB7 | 0.844 |
| Architecture | @top1* Imagenet| @top1* Noisy-Student|
| -------------- | :----: |:---:|
| EfficientNetB0 | 0.772 |0.788|
| EfficientNetB1 | 0.791 |0.815|
| EfficientNetB2 | 0.802 |0.824|
| EfficientNetB3 | 0.816 |0.841|
| EfficientNetB4 | 0.830 |0.853|
| EfficientNetB5 | 0.837 |0.861|
| EfficientNetB6 | 0.841 |0.864|
| EfficientNetB7 | 0.844 |0.869|

**\*** - topK accuracy score for converted models (imagenet `val` set)

Expand Down
2 changes: 1 addition & 1 deletion efficientnet/__version__.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.
# ==============================================================================
VERSION = (1, 0, 0)
VERSION = (1, 1, 0)

__version__ = ".".join(map(str, VERSION))

0 comments on commit 6ceca71

Please sign in to comment.