-
Notifications
You must be signed in to change notification settings - Fork 29
Conversation
|
||
## Model compression |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I don't understand why separate the rounding and pruning
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
how about set Deployment optimization methods
to Tools
and move the rounding to Model compression
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
As with the merge batch normalization, the rounding method is used to process the model before the inference. So I put it in the tool directory.
benchmark/README.md
Outdated
|
||
| Model | Xiaomi MI 5(ST) | | ||
|-----------| ----------- | | ||
| Mobilenet | 122.607(ms) | |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
need to specify the type of mobilenet (1.0 or 0.75 , 0.5 0.25) and input size(224* 224 * 3)
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Done.
benchmark/model/mobilenet.py
Outdated
name="output", input=net, size=classdim, act=paddle.activation.Softmax()) | ||
|
||
Inputs("image") | ||
Outputs("output") |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This config can not be used in v2 api, all mobilenet config in this repo should be the same specifications, and could use for v1 and v2 api, here is a example https://gist.github.com/NHZlX/204da5e89a1263ab6953a669401e37d5 , and this need to be discussed.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I temporarily remove this config file.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
No description provided.