GitXplorerGitXplorer
l

EfficientNet-PyTorch

public
3 stars
2 forks
0 issues

Commits

List of commits on branch master.
Verified
bb503b1c0e927fde81cee2008b80d94120e41f28

Update README.md

llessw2020 committed 5 years ago
Verified
3d9d1ef85ae7f342442be2c34c3cdb9e1ae1d0dd

Update README.md

llessw2020 committed 5 years ago
Unverified
fc846f0fb9f40677b28f2feddc1dea244b4d98f8

models 0-7 now loading; added fastprogress.py

llessw2020 committed 5 years ago
Unverified
25a85f18f819b8d934a06f32b98ea38c18d3c166

update 11-effNet - now working for B0

llessw2020 committed 5 years ago
Unverified
ff88becd8bf161cb2865754a5278eb4de4151e40

notebook for effNet

llessw2020 committed 5 years ago
Unverified
16ddbc80e739fe235168f336003762f273aaeeea

drop connect default to 0

llessw2020 committed 5 years ago

README

The README file for this repository.

EfficientNet-PyTorch

Unofficial port of Google's new EfficientNet to Pytorch and FastAI.

Models B0-7 loading.
Note that the code is in the notebook and assumes you have access to FastAI dev course 2 notebooks.
Will remove that dependency and port to .py file after further testing.

Usage:

arch = effNet(model=5, c_out = 10)
where model = B0-B7, and c_out = number of classes in classifier.

Here's the official code in TF:

https://github.com/tensorflow/tpu/blob/master/models/official/efficientnet/efficientnet_model.py

and info about it from their blog: https://ai.googleblog.com/2019/05/efficientnet-improving-accuracy-and.html