New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Subpackage nnabla.models #337

Merged

TE-AkioHayakawa merged 22 commits into master from 20190129-nnabla-models

Jan 29, 2019

Contributor

TE-TakuyaNarihira commented Jan 29, 2019

This adds nnabla.models API which allows users to use SotA pretrained models for both inference and training.

ImageNet pretrained models are added. Lineup so far is as following;

ResNet-18,34,50,101,152
MobileNetV2
SqueezeNet
SENet-154

We'll add more very soon.

NNabla developer and others added some commits Jan 29, 2019


      [format]

5548aef


      [python][docker] Modified SimpleGraph

SimpleGraph is now able to return graphviz digraph object that can be
used to display graph in Jupyter notebook.

Also, now parameter variable names are also drawn in the graph node.
The need_grad state is represented as black boundary in the variable
node.

f8bac7f


      [python][cli] fix return value in cli plot

50c07f9


      [python] Move download utilities to a module

05f42dc


      [python] Show download progress in Bytes

b1fed43


      [doc] Fix plot_time_elapsed typo

c91ecd3


      [python] NNP graph loader with pre-ordering functions

978977a


      [cli] Drawing graph with graphviz

3b7cd3f


      [python] parameter_scope takes local parameter scope dict

1b17bf4


      [python] filter out isolated variables from inputs and outputs

attributes in NnpNetwork

d72b6ec


      [python] Allow nntxt as an input in NnpLoader

c7db0d9


      [python] Fix bug due to unintended auto_forward in NnpNetwork

9fea574


      [python] Add variable name in NnpLoader

4e268a3


      [python] NnpLoader takes callbacks that manipulate network


      [python][models] Add ResNet

2b24a37


      [models] Add MobileNetV2

a027009


      [models] refactor imagenet model code and tests

f283181


      [models] Add SENet-154

35ed7f2


      [models] Add SqueezeNet v1.1

1f1755a


      [models] Add performance table for imagenet models

f630a61


      [python] Fix NnpLoader for Python2

9bd0c68


      [python][test] Fix and refactoring nnabla.models

- remove_and_rewire uses input name after rewiring
- Fix SENet pooling
- Refactoring

4a3bd9f

TE-TakuyaNarihira assigned TE-AkioHayakawa Jan 29, 2019

TE-AkioHayakawa merged commit `4ffc87d` into master Jan 29, 2019

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

You can’t perform that action at this time.